Функция PyTorch torch.promote_types

Pytorch torch 参考手册 Справочное руководство Pytorch torch

torch.promote_types— это функция в PyTorch, используемая для продвижения типов. Она принимает несколько типов данных на вход и возвращает общий тип (наиболее подходящий тип), который может вместить все входные типы.

Определение функции

torch.promote_types(type1, type2)

Описание параметра

type1: Первый тип данных
type2: Второй тип данных

Пример использования

Пример

import torch

# Пример продвижения типов
dtype1 = torch.float32
dtype2 = torch.float64

# Получить общий тип
result_dtype = torch.promote_types(dtype1, dtype2)

print("Общий тип float32 и float64:", result_dtype)

# Целочисленные и вещественные типы
dtype3 = torch.int32
dtype4 = torch.float32
result_dtype2 = torch.promote_types(dtype3, dtype4)

print("Общий тип int32 и float32:", result_dtype2)

Результат вывода:

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