Диапазон в PyTorch
У arange задают начало, конец
и шаг между соседними целыми.
Конечное значение в ряд не попадает:
диапазон обрывается прямо перед ним.
Если указать только конец, ряд начнётся с нуля. Создадим такой диапазон и выведем его:
import torch
vals = torch.arange(5)
print(vals) # выведет tensor([0, 1, 2, 3, 4])
Можно задать и начало, и конец.
Создадим ряд от 2 до 6
и выведем его:
import torch
vals = torch.arange(2, 6)
print(vals) # выведет tensor([2, 3, 4, 5])
Третьим аргументом идёт шаг.
Создадим ряд от 2 до 8
с шагом 2 и выведем его:
import torch
vals = torch.arange(2, 8, 2)
print(vals) # выведет tensor([2, 4, 6])
Шаг может быть отрицательным, тогда числа идут вниз. Создадим такой ряд и выведем его:
import torch
vals = torch.arange(10, 0, -3)
print(vals) # выведет tensor([10, 7, 4, 1])
Создайте ряд целых от 0 до
6 не включая конец и выведите его.
Создайте ряд чётных целых от 0
до 10 не включая конец
с шагом 2 и выведите его.
Создайте убывающий ряд от 9
до 0 не включая конец
с шагом -3 и выведите его.