Which variable would correspond to the depth map?
Generally you can use matplotlib.pyplot to display image-like arrays.
If you are dealing with tensors, you can transform it to a numpy array via tensor.numpy().