Получить тип данных тензора PyTorch

Question

Я понимаю, что тензоры PyTorch однородны, т. Е. Каждый из элементов одного типа.

Как узнать тип элементов в тензоре PyTorch?

Answer 1

Есть три вида вещей:

dtype                   || CPU tensor               || GPU tensor

torch.float32              torch.FloatTensor           torch.cuda.FloatTensor

Первый, который вы получите с print(t.dtype), если t - ваш тензор, иначе вы используете t.type() для двух других.

Answer 2

Используйте атрибут dtype :

>>> import torch
>>> print(torch.Tensor(1).dtype)
torch.float32

В документации также приведен канонический список типов данных .