PyTorch torch.promote_types Funktion
PyTorch torch Referenzhandbuch
torch.promote_typesEs ist eine Funktion in PyTorch, die für Typ-Promotion verwendet wird. Sie akzeptiert mehrere Datentypen als Eingabe und gibt den gemeinsamen Typ (den am besten geeigneten Typ) zurück, der alle Eingabetypen aufnehmen kann.
Funktionsdefinition
torch.promote_types(type1, type2)
Parameterbeschreibung
type1: Erster Datentyptype2: Zweiter Datentyp
Anwendungsbeispiel
Beispiel
import torch
# Beispiel für Typ-Promotion
dtype1 = torch.float32
dtype2 = torch.float64
# Gemeinsamen Typ ermitteln
result_dtype = torch.promote_types(dtype1, dtype2)
print("Gemeinsamer Typ von float32 und float64:", result_dtype)
# Integer- und Float-Typen
dtype3 = torch.int32
dtype4 = torch.float32
result_dtype2 = torch.promote_types(dtype3, dtype4)
print("Gemeinsamer Typ von int32 und float32:", result_dtype2)
# Beispiel für Typ-Promotion
dtype1 = torch.float32
dtype2 = torch.float64
# Gemeinsamen Typ ermitteln
result_dtype = torch.promote_types(dtype1, dtype2)
print("Gemeinsamer Typ von float32 und float64:", result_dtype)
# Integer- und Float-Typen
dtype3 = torch.int32
dtype4 = torch.float32
result_dtype2 = torch.promote_types(dtype3, dtype4)
print("Gemeinsamer Typ von int32 und float32:", result_dtype2)
Das Ausgabeergebnis ist:
float32 和 float64 的公共类型: torch.float64 int32 和 float32 的公共类型: torch.float32
Andere Erweiterungen