Как именно embedding_bag работает в PyTorch - PullRequest
0 голосов
/ 27 мая 2020

в PyTorch, torch.nn.functional.embedding_bag, кажется, является основной функцией, отвечающей за выполнение реальной работы по встраиванию поиска. В документации PyTorch упоминалось, что embedding_bag выполняет свою работу> без создания промежуточных вложений. Что именно это означает? Означает ли это, например, что в режиме «сумма» выполняется суммирование на месте? или это просто означает, что при вызове embedding_bag не будет создаваться никаких дополнительных тензоров, но все же с точки зрения системы все промежуточные векторы-строки уже загружены в процессор, который будет использоваться для вычисления окончательного тензора?

...