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

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

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

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

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

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

Пример

import torch

x = torch.randn(2, 3)
y = torch.ones_like(x)

print(y)

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

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

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

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