获取PyTorch张量的数据类型

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

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

使用：

---

文档还提供了。

有三种情况：

dtype                   || CPU tensor               || GPU tensor

torch.float32              torch.FloatTensor           torch.cuda.FloatTensor

---

如果

t

是你的张量，那么你用

print（t.dtype）

得到的第一个是你的张量，否则你用

t.type（）

得到另外两个。

你是如何得到类类型的？在Pytorch中有这样做的方法吗？@Breeze你能举例说明你的意思吗？

dtype                   || CPU tensor               || GPU tensor

torch.float32              torch.FloatTensor           torch.cuda.FloatTensor