image_output_1.png
import numpy as np

disp = np.zeros(shape=(16000, 40, 40))
sim = np.zeros(shape=(16000, 9, 40, 40, 1))
topo = np.zeros(shape=(16000, 1, 40, 40, 1))

img3 = sim[0][1].reshape(40, 40)
print(img3.shape)