'][0,0,:,:]);plt.show() # #plt.figure(0);plt.imshow(our_output[0][0,:,:,0]); #plt.figure(0);plt.imshow(torch_output[b'network/conv0_x/conv0'][0,0,:,:]);plt.show() # #plt.figure(0);plt.imshow(our_output[0][0,:,:,0]); #plt.figure(0);plt.imshow(torch_output[b'network/conv0_x/conv0_0/conv_0'][0,0,:,:]);plt.show() # #plt.figure(0);plt.imshow(our_output[0][0,:,:,0]); #plt.figure(0);plt.imshow(torch_output[b'network/conv0_x/conv0_0/conv_0'][0,0,:,:]);plt.show() 