Функция zeros

Функция zeros создает тензор, все элементы которого равны нулю. Первым параметром функция принимает форму тензора в виде кортежа или списка чисел. Вторым параметром можно передать тип данных элементов. Также можно указать устройство (CPU или GPU).

Синтаксис

torch.zeros(size, [dtype], [device])

Пример

Давайте создадим одномерный тензор из пяти нулей:

import torch

t = torch.zeros(5)
print(t)

Результат выполнения кода:

tensor([0., 0., 0., 0., 0.])

Пример

Создадим двумерный тензор размером 2 на 3, заполненный нулями:

import torch

t = torch.zeros((2, 3))
print(t)

Результат выполнения кода:

tensor([
	[0., 0., 0.],
	[0., 0., 0.],
])

Пример

Укажем тип данных элементов как целые числа:

import torch

t = torch.zeros((2, 2), dtype=torch.int)
print(t)

Результат выполнения кода:

tensor([
	[0, 0],
	[0, 0],
])

Смотрите также

функцию zeros_like,
которая создает тензор из нулей по форме другого тензора
функцию ones,
которая создает тензор, заполненный единицами
функцию empty,
которая создает тензор без инициализации значений
функцию full,
которая создает тензор, заполненный указанным значением