Fonction torch.chunk de PyTorch
Manuel de référence Pytorch torch
torch.chunkC'est une fonction de PyTorch utilisée pour diviser un tenseur en morceaux le long d'une dimension spécifiée.
Définition de fonction
torch.chunk(tensor, chunks, dim)
Exemple d'utilisation
Exemple
import torch
x = torch.arange(12).reshape(3, 4)
# Diviser en 3 morceaux
result = torch.chunk(x, 3, dim=0)
print("Nombre de morceaux :", len(result))
for i, t in enumerate(result):
print(f"Morceau {i} :", t.shape)
x = torch.arange(12).reshape(3, 4)
# Diviser en 3 morceaux
result = torch.chunk(x, 3, dim=0)
print("Nombre de morceaux :", len(result))
for i, t in enumerate(result):
print(f"Morceau {i} :", t.shape)
Autres extensions