PyTorch torch.broadcast_to Funktion
PyTorch torch Referenzhandbuch
torch.broadcast_toEs ist eine Funktion in PyTorch, die verwendet wird, um einen Tensor auf eine angegebene Form zu broadcasten. Sie gibt eine Ansicht des Eingabetensors zurück, die auf die Zielform gebroadcastet wird. Die Broadcasting-Regeln folgen dem Broadcasting-Mechanismus von NumPy.
Funktionsdefinition
torch.broadcast_to(input, shape)
Anwendungsbeispiel
Beispiel
import torch
# Grundlegende Verwendung: Einen 1-D-Tensor auf 2-D broadcasten
x = torch.tensor([1, 2, 3])
y = torch.broadcast_to(x, (3, 3))
print("Original:", x)
print("Nach dem Broadcasting:")
print(y)
# tensor([[1, 2, 3],
# [1, 2, 3],
# [1, 2, 3]])
# Einen Skalar auf eine größere Form broadcasten
x = torch.tensor(5)
y = torch.broadcast_to(x, (2, 3, 4))
print("Skalar auf (2,3,4) gebroadcastet:", y.shape)
# Einen 2-D-Tensor auf 3-D broadcasten
x = torch.tensor([[1, 2], [3, 4]]) # (2, 2)
y = torch.broadcast_to(x, (3, 2, 2))
print("Auf (3,2,2) gebroadcastet:", y.shape)
print(y)
# Grundlegende Verwendung: Einen 1-D-Tensor auf 2-D broadcasten
x = torch.tensor([1, 2, 3])
y = torch.broadcast_to(x, (3, 3))
print("Original:", x)
print("Nach dem Broadcasting:")
print(y)
# tensor([[1, 2, 3],
# [1, 2, 3],
# [1, 2, 3]])
# Einen Skalar auf eine größere Form broadcasten
x = torch.tensor(5)
y = torch.broadcast_to(x, (2, 3, 4))
print("Skalar auf (2,3,4) gebroadcastet:", y.shape)
# Einen 2-D-Tensor auf 3-D broadcasten
x = torch.tensor([[1, 2], [3, 4]]) # (2, 2)
y = torch.broadcast_to(x, (3, 2, 2))
print("Auf (3,2,2) gebroadcastet:", y.shape)
print(y)
Andere Erweiterungen