Fonction torch.range de PyTorch
Manuel de référence Pytorch torch
torch.rangeC'est une fonction de PyTorch utilisée pour créer un tenseur de séquence arithmétique qui inclut la valeur de fin.
Définition de fonction
torch.range(start=0, end, step=1, dtype=None, device=None, requires_grad=False)
Exemples d'utilisation
Exemple
import torch
# Créer une séquence arithmétique de 0 à 5 (5 inclus)
x = torch.range(5)
print(x)
# Créer une séquence arithmétique de 1 à 10 avec un pas de 2
y = torch.range(1, 10, 2)
print(y)
# Créer une séquence arithmétique de 0 à 5 (5 inclus)
x = torch.range(5)
print(x)
# Créer une séquence arithmétique de 1 à 10 avec un pas de 2
y = torch.range(1, 10, 2)
print(y)
Autres extensions