1.view()

直接上例子:

import torch

x=torch.randn(6,6)

y=x.view(x.size(0),2,-1)  #保持第一维,对后面两维做resize

y.size()=(6,2,3)

2.cat (多个tensors的拼接)

import torch

x=torch.randn(2,3)

torch.cat((x,x),0)  #行拼接4*3

torch.cat((x,x),1 or -1) #列拼接2*6

3.module

module是所有网络的基类,modules返回一个包含 当前模型 所有模块的迭代器,add_module将一个 child module 添加到当前 modle。 被添加的module可以通过 name属性来获取。ModuleList可以像一般的Python list一样被索引,例:

Pytorch常用函数

持续更新ing..........


相关文章:

  • 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-08-19
  • 2021-07-14
  • 2021-11-14
  • 2021-08-08
  • 2021-06-18
  • 2022-12-23
  • 2023-02-02
相关资源
相似解决方案