Función torch.typename de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.typenameEs una función de PyTorch utilizada para obtener el nombre del tipo de dato. Acepta un tensor o un tipo de dato como entrada y devuelve el nombre de cadena de su tipo.

Definición de la función

torch.typename(input)

Ejemplo de uso

Ejemplo

import torch

# Obtener el nombre del tipo del tensor
x = torch.tensor([1, 2, 3], dtype=torch.float32)
print("Tipo de tensor float32:", torch.typename(x))

# Obtener el nombre del tipo de dato
dtype = torch.float64
print("Tipo float64:", torch.typename(dtype))

# Tipo complejo
y = torch.tensor([1+2j], dtype=torch.complex64)
print("Tipo de tensor complex64:", torch.typename(y))

El resultado de salida es:

float32 张量类型: torch.FloatTensor
float64 类型: torch.DoubleTensor
complex64 张量类型: torch.ComplexFloatTensor

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones