import torch.nn as nn class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.linear = nn.Linear(10, 20)