gregschechter

Code example at https://pytorch.org/features/ is incorrect

1

## 📚 Documentation The first code shown in https://pytorch.org/features is ``` import torch class MyModule(torch.nn.Module): def __init__(self, N, M): super(MyModule, self).__init__() self.weight = torch.nn.Parameter(torch.rand(N, M)) def forward(self, input): if input.sum()...