Только что нашел я случайно кодировал some_data.unsqueeze (0) .to (device) и some_data.to (device) .unsqueeze (0) .
Если я правильно помню, torch.tensor.to ? /> Что заставляет меня задуматься, предположим, что устройство установлено на графический процессор, различный порядок между .to (device) и операция тензора дает различную производительность и т. Д.
Подробнее здесь: https://stackoverflow.com/questions/796 ... -operation
Some_operation (). to (device) vs to (device) .some_operation () ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение