Функция torch.range в PyTorch
Справочное руководство Pytorch torch
torch.rangeЭто функция в PyTorch, используемая для создания тензора арифметической последовательности, включающего конечное значение.
Определение функции
torch.range(start=0, end, step=1, dtype=None, device=None, requires_grad=False)
Примеры использования
Пример
import torch
# Создать арифметическую последовательность от 0 до 5 (включая 5)
x = torch.range(5)
print(x)
# Создать арифметическую последовательность от 1 до 10 с шагом 2
y = torch.range(1, 10, 2)
print(y)
# Создать арифметическую последовательность от 0 до 5 (включая 5)
x = torch.range(5)
print(x)
# Создать арифметическую последовательность от 1 до 10 с шагом 2
y = torch.range(1, 10, 2)
print(y)
Другие расширения