Pytorch 之torch.nn初探

第1关：torch.nn.Module

本关要求利用nn.Linear()声明一个线性模型 l，并构建一个变量 net 由三个l序列构成。

import torch
import torch.nn as nn
from torch.autograd import Variable
#/********** Begin *********/
#声明一个in_features=2,out_features=3的线性模型 l并输出
l = nn.Linear(2, 2)
print