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

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

torch.onesэто функция в PyTorch, используемая для создания тензора из единиц. Она создаёт тензор заданной формы, все элементы которого инициализированы значением 1.

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

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

torch.ones(*size, dtype=None, device=None, requires_grad=False, pin_memory=False)

Parameters:

  • *size(int): форма тензора, например,3、(3, 4)、(2, 3, 4)и т. д.
  • dtype(torch.dtype, необязательно): задаёт тип данных тензора, по умолчаниюtorch.float32。
  • device(torch.device, необязательно): задаёт устройство, на котором хранится тензор.
  • requires_grad(bool, необязательно): Вычислять ли градиенты.
  • pin_memory(bool, необязательно): использовать ли закреплённую память.

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

  • torch.Tensor: Возвращает тензор, заполненный единицами.

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

Пример 1: Создать одномерный тензор из единиц

Пример

import torch

# Создать тензор из единиц, содержащий 5 элементов
x = torch.ones(5)

print(x)

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

tensor([1., 1., 1., 1., 1.])

Пример 2: Создать двумерный тензор из единиц

Пример

import torch

# Создать тензор из единиц размера 3x4 (матрицу)
x = torch.ones(3, 4)

print(x)
print(x.shape)

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

tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
torch.Size([3, 4])

Пример 3: Создание 3D-тензора из единиц

Пример

import torch

# Создать тензор 2x3x4, заполненный единицами
x = torch.ones(2, 3, 4)

print(x.shape)

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

torch.Size([2, 3, 4])

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

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