Функция ones
Функция ones создает тензор, заполненный единицами.
Первым параметром функция принимает размерность тензора
в виде кортежа или списка чисел. Вторым параметром
можно указать тип данных элементов. Также доступны
дополнительные аргументы для выбора устройства
и необходимости вычисления градиента.
Синтаксис
torch.ones(size, [dtype], [device], [requires_grad])
Пример
Давайте создадим одномерный тензор из трех единиц:
import torch
t = torch.ones((3,))
print(t)
Результат выполнения кода:
tensor([1., 1., 1.])
Пример
Создадим двумерный тензор размером 2 на 3:
import torch
t = torch.ones((2, 3))
print(t)
Результат выполнения кода:
tensor([
[1., 1., 1.],
[1., 1., 1.],
])
Пример
Укажем тип данных int при создании тензора:
import torch
t = torch.ones((2, 2), dtype=torch.int64)
print(t)
Результат выполнения кода:
tensor([
[1, 1],
[1, 1],
], dtype=torch.int64)
Пример
Создадим трехмерный тензор и проверим его размерность:
import torch
t = torch.ones((2, 3, 4))
print(t.shape)
Результат выполнения кода:
torch.Size([2, 3, 4])
Пример
Создадим тензор, для которого будет вычисляться градиент:
import torch
t = torch.ones((3,), requires_grad=True)
print(t)
print(t.requires_grad)
Результат выполнения кода:
tensor([1., 1., 1.], requires_grad=True)
True
