PyTorch torch.promote_types 関数
Pytorch torch リファレンスマニュアルtorch.promote_typesPyTorch における型昇格に使用される関数です。複数のデータ型を入力として受け取り、すべての入力型を格納できる共通型（最適な型）を返します。
関数定義
torch.promote_types(type1, type2)
パラメータの説明
type1: 最初のデータ型type2: 2番目のデータ型
使用例
例
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)
# 型昇格の例
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
その他の拡張