File size: 138 Bytes
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)