Pytorch常用函数
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
一样被索引,例:
持续更新ing..........