Функция PyTorch torch.zeros_like

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

torch.zeros_likeэто функция в PyTorch, используемая для создания тензора, заполненного нулями, с той же формой, что и входной тензор.

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

torch.zeros_like(input, dtype=None, device=None, requires_grad=False)

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

Пример

import torch

x = torch.randn(3, 4)
y = torch.zeros_like(x)

print(форма x:, x.shape)
print(форма y:, y.shape)
print(y)

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

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