如何查看一个tensor向量是long还是float的类型
可以通过tensor的dtype属性来查看一个tensor向量的数据类型。dtype属性返回的是一个torch.dtype对象,可以使用其方法来判断数据类型。下面是一个例子:
import torch
# 创建一个long类型的tensor向量
tensor_long = torch.tensor([1, 2, 3], dtype=torch.long)
print(tensor_long.dtype) # 输出 torch.int64
# 创建一个float类型的tensor向量
tensor_float = torch.tensor([1.0, 2.0, 3.0], dtype=torch.float)
print(tensor_float.dtype) # 输出 torch.float32
在上面的例子中,通过tensor的dtype属性可以分别判断tensor_long和tensor_float的数据类型是long还是float。
原文地址: http://www.cveoy.top/t/topic/javN 著作权归作者所有。请勿转载和采集!