Displaying the depth map


I am new to programming with python. How can I display the depth map images using this code?

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().