Fonction torch.promote_types de PyTorch

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

torch.promote_typesIl s'agit d'une fonction dans PyTorch utilisée pour la promotion de types. Elle accepte plusieurs types de données en entrée et renvoie le type commun (le type le plus approprié) pouvant accueillir tous les types d'entrée.

Définition de fonction

torch.promote_types(type1, type2)

Description du paramètre

  • type1: Premier type de données
  • type2: Deuxième type de données

Exemple d'utilisation

Exemple

import torch

# Exemple de promotion de type
dtype1 = torch.float32
dtype2 = torch.float64

# Obtenir le type commun
result_dtype = torch.promote_types(dtype1, dtype2)

print("Type commun de float32 et float64 :", result_dtype)

# Types entiers et flottants
dtype3 = torch.int32
dtype4 = torch.float32
result_dtype2 = torch.promote_types(dtype3, dtype4)

print("Type commun de int32 et float32 :", result_dtype2)

Le résultat de sortie est:

float32 和 float64 的公共类型: torch.float64
int32 和 float32 的公共类型: torch.float32

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

Autres extensions