今天小编就为大家分享一篇pytorch 在sequential中使用view来reshape的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

pytorch中view是tensor方法，然而在sequential中包装的是nn.module的子类， 因此需要自己定义一个方法： import torch.nn as nn class Reshape(nn.Module): def __init__(self, *args): super(Reshape, self).__init__() self.shape = args def forward(self, x): # 如果数据集最后一个batch样本数量小于定义的b