PyTorch torch.chunk Funktion
PyTorch torch Referenzhandbuch
torch.chunkEs ist eine Funktion in PyTorch, die verwendet wird, um einen Tensor entlang einer angegebenen Dimension in Chunks aufzuteilen.
Funktionsdefinition
torch.chunk(tensor, chunks, dim)
Anwendungsbeispiel
Beispiel
import torch
x = torch.arange(12).reshape(3, 4)
# In 3 Chunks aufteilen
result = torch.chunk(x, 3, dim=0)
print("Anzahl der Chunks:", len(result))
for i, t in enumerate(result):
print(f"Chunk {i}:", t.shape)
x = torch.arange(12).reshape(3, 4)
# In 3 Chunks aufteilen
result = torch.chunk(x, 3, dim=0)
print("Anzahl der Chunks:", len(result))
for i, t in enumerate(result):
print(f"Chunk {i}:", t.shape)
Andere Erweiterungen