Функция torch.arange в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.arange— это функция в PyTorch, используемая для создания тензоров арифметической прогрессии. Она создает одномерный тензор, содержащий арифметическую последовательность от начального значения до конечного значения.

Это часто используется в глубоком обучении для создания индексов, массивов диапазонов или генерации последовательностей в циклах.

Определение функции

torch.arange(start=0, end, step=1, dtype=None, device=None, requires_grad=False)

Parameters:

  • start(float, необязательно): Начальное значение последовательности, по умолчанию 0.
  • end(float): Конечное значение последовательности (не включительно).
  • step(float, необязательно): Шаг, по умолчанию 1.
  • dtype(torch.dtype, необязательно): Указывает тип данных тензора.
  • device(torch.device, необязательно): Указывает устройство, на котором хранится тензор.
  • requires_grad(bool, необязательно): Вычислять ли градиенты.

Возвращаемое значение:

  • torch.Tensor: Возвращает одномерный тензор.

Примеры использования

Пример 1: От 0 до 5

Пример

import torch

# Создать арифметическую последовательность от 0 до 4
x = torch.arange(5)

print(x)

Результат вывода:

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

Пример 2: Указание начального и конечного значений

Пример

import torch

# Создать арифметическую последовательность от 2 до 8
x = torch.arange(2, 9)

print(x)

Результат вывода:

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

Пример 3: Указание размера шага

Пример

import torch

# Создать арифметическую последовательность от 0 до 10 с размером шага 2
x = torch.arange(0, 11, 2)

print(x)

Результат вывода:

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

Пример 4: Отрицательный размер шага

Пример

import torch

# Создать арифметическую последовательность от 10 до 0 с размером шага -2
x = torch.arange(10, 0, -2)

print(x)

Результат вывода:

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

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения