获取 PyTorch 张量的数据类型

Question

我了解 PyTorch 张量是同质的，即每个元素都属于同一类型。

如何找出 PyTorch 张量中元素的类型？

Answer 1

使用dtype attribute：

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

文档还给出了canonical list of datatypes。

Answer 2

一共有三种：

dtype                   || CPU tensor               || GPU tensor

torch.float32              torch.FloatTensor           torch.cuda.FloatTensor

如果t 是您的张量，则使用print(t.dtype) 获得的第一个，否则您使用t.type() 作为其他两个。