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以后的版本

持续更新