how to do I print model summary like what we do in keras model.summary().
Also I am using conv1d and I need to use the output of this to conv2d for which I need to add to reshape the output of conv1d. How to do that?
print (model) itself will give a summary of it’s properly represented in network class.
Torch. Reshape or view can be used to convert 1d output to 2d output.