running in jupyter notebook, it shows that code of defining network is OK, error occurs in the statement of feedforward as follows:
input = torch.randn(1, 1, 32, 32)
(error)---------->out = net(input)
print(out)