torch.arange(): Generarea Secvențelor de Numere cu Pas Constant în PyTorch & Deep Learning Practic

Funcția torch.arange() returnează un tensor 1D conținând valori echidistante în intervalul semi-deschis [start, end), incrementate cu pasul specificat step. Reprezintă echivalentul direct al funcției range() din Python sau np.arange() din NumPy.

Exemplu de utilizare

import torch

# Secvență de la 0 la 10 cu pasul 2
indici = torch.arange(0, 10, step=2)
print(indici)  # tensor([0, 2, 4, 6, 8])

# Poziții de tokeni pentru embedding-uri poziționale
pozitii = torch.arange(512)

Utilizată intensiv pentru generarea indexărilor temporale și spațiale în arhitectura Transformer.