直接将参数传入即可。如下代码，有三个以上的输入也是以此类推
summary(model,first_input,second_input,device='cpu')
还有问题可以去这里查看torch-summary