ad2100bf40e81c9964b9f4c8a3e5a9dbf5f4efd9,tests/layers/view.py,TestView,forward,#TestView#,16

Before Change

    def forward(self, x):
        x = self.conv2d(x)
        x = x.view([x.size(0), -1, 2, 1, 1, 1, 1, 1]).view(x.size(0), -1).view(x.size(0), -1)
        x = torch.nn.Tanh()(x)
        return x

After Change

        print(type(x.size()[0]))

        x = x.view([int(x.size(0)), -1])
        x = self.fc(x)
        return x

Instances