为什么PyTorch收集函数要求索引参数为LongTensor类型？

Question

我正在PyTorch中编写一些代码，并且遇到了collect函数。查看文档，我看到index参数包含LongTensor，为什么呢？为什么需要使用LongTensor代替其他类型（如IntTensor）？有什么好处？

Answer 1

默认情况下，pytorch中的所有索引都表示为长张量-允许索引超过4GB元素（“ regular” int的最大值）的非常大的张量。