今天小编就为大家分享一篇pytorch中获取模型input/output shape实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

Pytorch官方目前无法像tensorflow, caffe那样直接给出shape信息，详见 https://github.com/pytorch/pytorch/pull/3043 以下代码算一种workaround。由于CNN, RNN等模块实现不一样，添加其他模块支持可能需要改代码。 例如RNN中bias是bool类型，其权重也不是存于weight属性中，不过我们只关注shape够用了。 该方法必须构造一个输入调用forward后（model(x)调用）才可获取shape #codi