summaryrefslogtreecommitdiff
path: root/test/data/network1.py
blob: 68fbe3769650cce63020c44abf0c73230748373f (plain)
1
2
3
4
5
6
7
8
import torch.nn as nn


class Net(nn.Module):

    def __init__(self):
        super(Net, self).__init__()
        self.linear = nn.Linear(10, 20)