Что означает -1 в тензор.размер (-1)?

Question

Я видел что-то подобное в документации Pytorch,

import torch

a = torch.tensor([1, 2])
a.size() # torch.Size([2])
a.size(-1) # 2

Как это работает?Я не нашел описание.Спасибо,

Answer 1

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)

Надеюсь, это поможет вам.