Fonction torch.result_type de PyTorch

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

torch.result_typeest une fonction de PyTorch utilisée pour déterminer le type de résultat d'une opération. Elle accepte des tenseurs ou des scalaires en entrée et renvoie le type de données à utiliser après avoir effectué l'opération concernée.

Définition de fonction

torch.result_type(tensor, scalar)

Description du paramètre

  • tensor: tenseur d'entrée ou type de données
  • scalar: valeur scalaire ou un autre tenseur

Exemple d'utilisation

Exemple

import torch

# Créer des tenseurs de types différents
a = torch.tensor([1, 2, 3], dtype=torch.float32)
b = torch.tensor([4, 5, 6], dtype=torch.float64)

# Obtenir le type de résultat
result_dtype = torch.result_type(a, b)

print("Type de résultat de float32 et float64 :", result_dtype)

# Utilisation d'un scalaire
c = torch.tensor([1, 2, 3], dtype=torch.int32)
result_dtype2 = torch.result_type(c, 1.5)

print("Type de résultat de int32 et 1.5 :", result_dtype2)

Le résultat de sortie est:

float32 和 float64 的结果类型: torch.float64
int32 和 1.5 的结果类型: torch.float32

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

Autres extensions