Fonction torch.typename de PyTorch

Pytorch torch 参考手册 Manuel de référence PyTorch torch

torch.typenameIl s'agit d'une fonction de PyTorch utilisée pour obtenir le nom du type de données. Elle accepte un tenseur ou un type de données en entrée et renvoie le nom de son type sous forme de chaîne.

Définition de fonction

torch.typename(input)

Exemple d'utilisation

Exemple

import torch

# Obtenir le nom du type du tenseur
x = torch.tensor([1, 2, 3], dtype=torch.float32)
print("type de tenseur float32:", torch.typename(x))

# Obtenir le nom du type de données
dtype = torch.float64
print("type float64 :", torch.typename(dtype))

# Type complexe
y = torch.tensor([1+2j], dtype=torch.complex64)
print("type de tenseur complex64 :", torch.typename(y))

Le résultat de sortie est:

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

Pytorch torch 参考手册 Manuel de référence PyTorch torch

Autres extensions