Функция PyTorch torch.typename
Справочное руководство PyTorch torch
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
Другие расширения