deep-learning - 使用 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：索引 578437695752307201 超出尺寸 16 的维度 1

我究竟做错了什么？

标签： deep-learningpytorch

Answer 1

我遇到了类似的问题。它似乎是 pytorch 中的一个错误。