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

how does it work if I put the following inputs with the following shapes into my model? 350 is the batch_size and the 850 is just an integer.

input_a: torch.Size([350, 6556])
input_b: torch.Size([350, 6556])
input_c: torch.Size([350])
input_d: torch.Size([20, 6556])
input_e: 850