Model summary and reshaping the output of one layer to feed to next layer

(Kaushal Aggarwal) #1

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?

(balamurali) #2

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.

(Shubham Chandel) #3