Публикации по теме 'pytorch'


Понимание индексирования с помощью pytorch gather
Вы когда-нибудь пробовали использовать функцию сборки pytorch? Я сделал это, и это было очень сложно. Сама функция довольно полезна, но понять, как ее использовать, может быть сложно. Итак, какова цель функции сбора? Документы говорят: torch.gather ( input , dim , index , out=None , sparse_grad=False ) → Tensor Gathers values along an axis specified by dim. Итак, он собирает значения по оси. Но чем это отличается от обычной индексации? При использовании оператора [] вы..