Your answer is greatly appreciated, it solved my problem.Thank you very much! And i also want to ask, if I want to see the output size in the middle of the network, is there a better way? What I know so far is using net.named_children () get the ouput after layer by layer calculation, another way is manually calculate the size