-1在tensor.size(-1)中意味着什么?

时间:2018-10-12 04:59:42

标签: python pytorch

我在Pytorch文档中看到了类似的内容,

hashtags

这是如何工作的?我没有找到描述。谢谢,

1 个答案:

答案 0 :(得分:3)

a.size(-1)指的是最后一个尺寸。例如,如果x的形状为(10,20),则x.size(-1)指的是第二维,即20。 看一下以下示例:

import torch
a= torch.zeros((2,5)) # a is matrix of 2 rows and 5 columns all elements are 0
#size gives a 1d tensor containing the shapes
a.size(-1)# refers to the last element in the tensor

这等效于:

a_size= a.size()
a_size(-1)

希望这对您有所帮助。