我在Pytorch文档中看到了类似的内容,
hashtags
这是如何工作的?我没有找到描述。谢谢,
答案 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)
希望这对您有所帮助。