Описание проблемы:

Я сталкиваюсь с проблемой несовпадения типов данных между выходом модели и метками при вычислении функции потерь. В зависимости от того, в какой тип я изменяю тензоры (долгое или плавающее), возникают ошибки.

Имеющиеся тензоры имеют форму (42000, 1, 28, 28) для изображений...

Я новичок в PyTorch. Я прочитал много кода, который активно использует член тензора .data. Однако, когда я ищу .data в официальной документации и Google, то нахожу очень мало информации. Я предполагаю, что .data содержит данные в тензоре, но не совсем понимаю, когда мы его используем, а когда...

Как проверить, использует ли PyTorch GPU? Команда nvidia-smi может обнаружить активность GPU, но я хочу проверить это непосредственно из Python-скрипта.