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一样被索引,例:

Pytorch常用函数

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