PyTorch torch.typename 函数
torch.typename 是 PyTorch 中用于获取数据类型名称的函数。它接受一个张量或数据类型作为输入,返回其类型的字符串名称。
函数定义
torch.typename(input)
使用示例
实例
import torch
# 获取张量的类型名称
x = torch.tensor([1, 2, 3], dtype=torch.float32)
print("float32 张量类型:", torch.typename(x))
# 获取数据类型的类型名称
dtype = torch.float64
print("float64 类型:", torch.typename(dtype))
# 复数类型
y = torch.tensor([1+2j], dtype=torch.complex64)
print("complex64 张量类型:", torch.typename(y))
# 获取张量的类型名称
x = torch.tensor([1, 2, 3], dtype=torch.float32)
print("float32 张量类型:", torch.typename(x))
# 获取数据类型的类型名称
dtype = torch.float64
print("float64 类型:", torch.typename(dtype))
# 复数类型
y = torch.tensor([1+2j], dtype=torch.complex64)
print("complex64 张量类型:", torch.typename(y))
输出结果为:
float32 张量类型: torch.FloatTensor float64 类型: torch.DoubleTensor complex64 张量类型: torch.ComplexFloatTensor

Pytorch torch 参考手册