torch.ones_like()

TORCH.ONES_LIKE

torch.ones_like(input, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format) → Tensor

返回一个填充了标量值1的张量，shape 与 input 相同

torch.ones_like(input)相当于torch.ones(input)

例子

>>> input = torch.empty(2, 3)
>>> torch.ones_like(input)

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