获取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

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