torch.arange

torch.arange(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

Возвращает 1-мерный тензор размера $\left\lceil \frac{\text{end} - \text{start}}{\text{step}} \right\rceil$ со значениями из интервала [start, end) с общим шагом step, начиная с start.

Обратите внимание, что нецелые step подвержены ошибкам округления с плавающей точкой при сравнении с end; чтобы избежать несоответствий, мы рекомендуем вычесть небольшую эпсилон из end в таких случаях.

$$\text{out}_{{i+1}} = \text{out}_{i} + \text{step}$$

Параметры

• start (Число) – начальное значение для набора точек. По умолчанию: 0.
• end (Число) – конечное значение для набора точек
• step (Число) – разрыв между каждой парой смежных точек. По умолчанию: 1.

Ключевые аргументы

• out (Тензор, необязательно) – выходной тензор.
• dtype (torch.dtype, optional) – желаемый тип данных возвращаемого тензора. По умолчанию: если None, используется глобальный параметр по умолчанию (см. torch.set_default_tensor_type()). Если dtype не указан, тип данных выводится из других входных аргументов. Если любой из start, end, или stop является числом с плавающей точкой, тип данных dtype выводится по умолчанию, см. get_default_dtype(). В противном случае тип данных dtype выводится как torch.int64.
• layout (torch.layout, optional) – желаемый макет возвращаемого тензора. По умолчанию: torch.strided.
• device (torch.device, optional) – желаемое устройство возвращаемого тензора. По умолчанию: если None, используется текущее устройство для типа тензора по умолчанию (см. torch.set_default_tensor_type()). device будет ЦП для тензоров ЦП и текущим устройством CUDA для тензоров CUDA.
• requires_grad (bool, необязательно) – Если autograd должен записывать операции с возвращаемым тензором. По умолчанию: False.

Пример:

>>> torch.arange(5)
tensor([ 0,  1,  2,  3,  4])
>>> torch.arange(1, 4)
tensor([ 1,  2,  3])
>>> torch.arange(1, 2.5, 0.5)
tensor([ 1.0000,  1.5000,  2.0000])