PyTorch torch.split-Funktion
PyTorch torch-Referenzhandbuch
torch.splitEs ist eine Funktion in PyTorch, die verwendet wird, um einen Tensor entlang einer angegebenen Dimension aufzuteilen.
Funktionsdefinition
torch.split(tensor, split_size, dim)
Anwendungsbeispiel
Beispiel
import torch
x = torch.arange(12).reshape(3, 4)
# Entlang dim=0 aufteilen
result = torch.split(x, 1, dim=0)
for i, t in enumerate(result):
print(f"Teil {i}:")
print(t)
x = torch.arange(12).reshape(3, 4)
# Entlang dim=0 aufteilen
result = torch.split(x, 1, dim=0)
for i, t in enumerate(result):
print(f"Teil {i}:")
print(t)
Andere Erweiterungen