PyTorch torch.broadcast_shapes Funktion
PyTorch torch Referenzhandbuch
torch.broadcast_shapesEs ist eine Funktion in PyTorch, mit der das Ergebnis des Broadcastings mehrerer Shapes berechnet wird. Sie gibt die Form zurück, die sich nach dem Broadcasting der Eingabe-Shapes ergibt, ohne dabei tatsächlich Tensoren zu erzeugen.
Funktionsdefinition
torch.broadcast_shapes(*shapes)
Anwendungsbeispiel
Beispiel
import torch
# Berechne Shape-Broadcast-Ergebnis
result = torch.broadcast_shapes((3, 1), (1, 4), (3, 4))
print("(3,1) + (1,4) + (3,4) ->", result)
# Ausgabe: (3, 4)
# Mehrere Shapes
result = torch.broadcast_shapes((5,), (1, 5), (3, 1, 5))
print("(5,) + (1,5) + (3,1,5) ->", result)
# Ausgabe: (3, 1, 5)
# Einzelne Form
result = torch.broadcast_shapes((2, 3))
print("(2,3) ->", result)
# Ausgabe: (2, 3)
# Ein Fehler wird gemeldet, wenn Broadcasting nicht möglich ist
try:
result = torch.broadcast_shapes((3,), (4,))
except RuntimeError as e:
print("Broadcast-Fehler:", e)
# Berechne Shape-Broadcast-Ergebnis
result = torch.broadcast_shapes((3, 1), (1, 4), (3, 4))
print("(3,1) + (1,4) + (3,4) ->", result)
# Ausgabe: (3, 4)
# Mehrere Shapes
result = torch.broadcast_shapes((5,), (1, 5), (3, 1, 5))
print("(5,) + (1,5) + (3,1,5) ->", result)
# Ausgabe: (3, 1, 5)
# Einzelne Form
result = torch.broadcast_shapes((2, 3))
print("(2,3) ->", result)
# Ausgabe: (2, 3)
# Ein Fehler wird gemeldet, wenn Broadcasting nicht möglich ist
try:
result = torch.broadcast_shapes((3,), (4,))
except RuntimeError as e:
print("Broadcast-Fehler:", e)
Andere Erweiterungen