Caffe2 - Python API
A deep learning, cross platform ML framework
network2.py
1 import torch.nn as nn
2 
3 
4 class Net(nn.Module):
5 
6  def __init__(self):
7  super(Net, self).__init__()
8  self.linear = nn.Linear(10, 20)
9  self.relu = nn.ReLU()