torch 中的setNumInputDims()函数

最新推荐文章于 2022-07-09 16:11:43 发布

咕噜咕噜酱紫

最新推荐文章于 2022-07-09 16:11:43 发布

阅读量511

点赞数 1

分类专栏： torch 文章标签： torch setNumInputDims

本文链接：https://blog.csdn.net/u010741000/article/details/90201204

版权

torch 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

输入：

minibatch = torch.Tensor(5, 2, 3, 5)

m = nn.View(-1):setNumInputDims(1)

m:forward(minibatch)

----------------------------------------------------------------------------------------------------------------------------------------------------------------------------

输出：

30*5的tensor

-------------------------------------------------------------------------------------------------------------------------------------------------------------------------

输入：

minibatch = torch.Tensor(5, 2, 3, 5)

m = nn.View(-1):setNumInputDims(2)

m:forward(minibatch)

-------------------------------------------------------------------------------------------------------------------------------------------------------------------------

输出：

10*15的tensor

-------------------------------------------------------------------------------------------------------------------------------------------------------------------------

代码解读：

view(-1)的含义就是，给定一个tensor你知道你想要多少列，但是不知道想要多少行的时候，就用view(-1)。

setNumInputDims(n)的意思就是，把前面几个维度乘到一起，作为tensor的行数，其余几个维度的成乘积作为tensor的列数。所以上式中的setNumInputDims(2)表示的就是把（5,2,3,5）中的前两维作为新tensor的行数，所以setNumInputDims(2)获得的新tensor大小为10*15。setNumInputDims(1)获得的就是5*30的tensor。setNumInputDims(3)获得的就是30*5的tensor。