DEBSOURCES
Skip Quicknav
sources / pytorch / 1.7.1-7 / torch / testing / _internal / data / network2.py
123456789
import torch.nn as nn class Net(nn.Module): def __init__(self): super().__init__() self.linear = nn.Linear(10, 20) self.relu = nn.ReLU()