Fonction torch.split de PyTorch
Manuel de référence PyTorch torch
torch.splitIl s'agit d'une fonction dans PyTorch utilisée pour diviser un tenseur selon une dimension spécifiée.
Définition de fonction
torch.split(tensor, split_size, dim)
Exemple d'utilisation
Exemple
import torch
x = torch.arange(12).reshape(3, 4)
# Diviser selon dim=0
result = torch.split(x, 1, dim=0)
for i, t in enumerate(result):
print(f"Partie {i} :")
print(t)
x = torch.arange(12).reshape(3, 4)
# Diviser selon dim=0
result = torch.split(x, 1, dim=0)
for i, t in enumerate(result):
print(f"Partie {i} :")
print(t)
Autres extensions