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

持续更新

 

相关文章:

  • 2021-07-10
  • 2021-05-11
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-07-02
  • 2022-12-23
  • 2022-12-23
猜你喜欢
  • 2021-09-27
  • 2021-07-14
  • 2021-11-14
  • 2021-08-08
  • 2021-06-18
  • 2022-12-23
  • 2023-02-02
相关资源
相似解决方案