PyTorch torch.ones_like 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.ones_likePyTorch で入力テンソルと同じ形状の全1テンソルを作成するために使用される関数です。

関数定義

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 リファレンスマニュアル

その他の拡張