Почему функция сбора PyTorch требует, чтобы индексный аргумент имел тип LongTensor?

Question

Я пишу некоторый код на PyTorch, и я наткнулся на функцию сбора. Проверяя документацию, я увидел, что аргумент index принимает LongTensor, почему так? Почему необходимо использовать LongTensor вместо другого типа, например IntTensor? Каковы преимущества?

Answer 1

По умолчанию все индексы в pytorch представлены в виде длинных тензоров, что позволяет индексировать очень большие тензоры, превышающие всего 4 ГБ (максимальное значение типа «обычный» int).