9dd3461
1
2
3
4
5
6
7
8
9
import torch.nn as nn class Net(nn.Module): def __init__(self): super().__init__() self.linear = nn.Linear(10, 20)