Функция arange

Функция arange создаёт одномерный тензор, содержащий последовательность чисел в указанном диапазоне. Первым параметром передаётся начальное значение (по умолчанию 0), вторым - конечное значение (не включается в последовательность), третьим - шаг (по умолчанию 1). Также можно указать тип данных элементов через параметр dtype.

Синтаксис

torch.arange(stop, [dtype])
torch.arange(start, stop, [step], [dtype])

Пример

Давайте создадим последовательность чисел от 0 до 9:

import torch

t = torch.arange(10)
print(t)

Результат выполнения кода:

tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

Пример

Давайте создадим последовательность от 5 до 14:

import torch

t = torch.arange(5, 15)
print(t)

Результат выполнения кода:

tensor([5, 6, 7, 8, 9, 10, 11, 12, 13, 14])

Пример

Давайте создадим последовательность от 0 до 9 с шагом 2:

import torch

t = torch.arange(0, 10, 2)
print(t)

Результат выполнения кода:

tensor([0, 2, 4, 6, 8])

Пример

Давайте создадим последовательность с отрицательным шагом от 10 до 1:

import torch

t = torch.arange(10, 1, -2)
print(t)

Результат выполнения кода:

tensor([10, 8, 6, 4, 2])

Пример

Давайте создадим последовательность чисел с плавающей точкой от 0 до 1 с шагом 0.2:

import torch

t = torch.arange(0, 1.2, 0.2)
print(t)

Результат выполнения кода:

tensor([0.0000, 0.2000, 0.4000, 0.6000, 0.8000, 1.0000])

Пример

Давайте укажем тип данных элементов:

import torch

t = torch.arange(5, dtype=torch.float)
print(t)

Результат выполнения кода:

tensor([0., 1., 2., 3., 4.])

Смотрите также

функцию linspace,
которая создаёт последовательность с фиксированным количеством точек
функцию logspace,
которая создаёт последовательность в логарифмической шкале
функцию range,
которая создаёт последовательность с возможностью работы с типами
функцию zeros,
которая создаёт тензор, заполненный нулями