torch.zeros

torch.zeros(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

Возвращает тензор, заполненный скалярным значением 0, с формой, определенной переменным аргументом size.

Параметры:

size (int...) – последовательность целых чисел, определяющая форму выходного тензора. Может быть переменным числом аргументов или коллекцией, например, списком или кортежем.

Ключевые аргументы:

• out (Тензор, необязательно) – выходной тензор.
• dtype (torch.dtype, optional) – желаемый тип данных возвращаемого тензора. По умолчанию: если None, используется глобальное значение по умолчанию (см. torch.set_default_tensor_type()).
• layout (torch.layout, optional) – желаемая компоновка возвращаемого тензора. По умолчанию: torch.strided.
• device (torch.device, optional) – желаемое устройство возвращаемого тензора. По умолчанию: если None, используется текущее устройство для типа тензора по умолчанию (см. torch.set_default_tensor_type()). device будет ЦП для тензоров ЦП и текущее CUDA-устройство для CUDA-тензоров.
• requires_grad (bool, необязательно) – Если autograd должен записывать операции над возвращаемым тензором. По умолчанию: False.

Пример:

>>> torch.zeros(2, 3)
tensor([[ 0.,  0.,  0.],
        [ 0.,  0.,  0.]])

>>> torch.zeros(5)
tensor([ 0.,  0.,  0.,  0.,  0.])