pytorch常用函数
model相关:
1 nn.conv2d(in,out,ksize)
2 nn.linear(in,out,bias=True/False)
3 nn.Maxpool2d(ksize)
4 nn.Avgpool2d(ksize)
5 nn.softmax(a,dim=0/1,dtype=torch.float)
6 torch.stack([a,b,c,d],0)
7 torch.stack([a,b,c,d],1)
8 torch.Tensor() #相当于自动添加了Variable,当然,这是在torch0.4.0以后的版本
持续更新