netrans/bin/torch/testing/_internal/data/network1.py

9 lines
147 B
Python

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