Función torch.promote_types de PyTorch

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

torch.promote_typesEs una función de PyTorch utilizada para la promoción de tipos. Acepta múltiples tipos de datos como entrada y devuelve el tipo común (el tipo más adecuado) que puede contener todos los tipos de entrada.

Definición de la función

torch.promote_types(type1, type2)

Descripción de parámetros

  • type1: primer tipo de datos
  • type2: segundo tipo de datos

Ejemplo de uso

Ejemplo

import torch

# Ejemplo de promoción de tipos
dtype1 = torch.float32
dtype2 = torch.float64

# Obtener el tipo común
result_dtype = torch.promote_types(dtype1, dtype2)

print("El tipo común de float32 y float64:", result_dtype)

# Tipos enteros y tipos de punto flotante
dtype3 = torch.int32
dtype4 = torch.float32
result_dtype2 = torch.promote_types(dtype3, dtype4)

print("El tipo común de int32 y float32:", result_dtype2)

El resultado de salida es:

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

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

Otras extensiones