Pytorch wrote a neural network, input as many images, the output for a picture, I want to know me now predict the pictures above each data point is by which data input! I know that the neural network can contain all depth is deep, but I want to draw intuitive representation! Thank you for your bosses! Have ready-made code is best, I on the basis of the change, if there are no words can explain the thinking of writing code?