Model summary in pytorch

Question

Is there any way, I can print the summary of a model in PyTorch like model.summary() method does in Keras as follows?

Model Summary:
___________         


        

Answer 1

You can just use x.shape, in order to measure tensor's x dimensions