获取 PyTorch 张量的数据类型

Question

我知道 PyTorch 张量是同质的，即每个元素都是相同的类型。

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

Answer 1

事情分为三种：

dtype                   || CPU tensor               || GPU tensor

torch.float32              torch.FloatTensor           torch.cuda.FloatTensor

你得到的第一个print(t.dtype)ift是你的张量，否则你使用t.type()其他两个。

Answer 2

使用dtype属性：

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

该文档还提供了数据类型的规范列表。