Ошибка индекса вне границ при использовании Pytorch

Question

У меня есть Два тензора

Я пытаюсь собрать по одному из каждой строки со столбцом, указанным этими индексами. Поэтому я пытаюсь получить:

[0,0,0,0,0,0,0,0,0,0,0,0,0,0,1,1]

Вот мой код для этого:

self.manDistMat.gather(1, state.unsqueeze(-1)))
self.manDistMat

- это матрица 16x16, а state.unsqueeze(-1) - другая матрица.

Когда я пытаюсь это сделать, я получаю эту ошибку.

RuntimeError: index 578437695752307201 выходит за пределы для измерения 1 с размером 16

Что я делаю неправильно?

Answer 1

Я на самом деле понял, что это потому, что я индексировал с тензором uint8. Когда я переключился на .long (), это сработало. Кто-нибудь может объяснить, почему это должен быть длинный тензор?