PyTorch torch.randn_like 関数
Pytorch torch リファレンスマニュアルtorch.randn_likePyTorchで入力テンソルと同じ形状の標準正規分布に従うランダムテンソルを作成するための関数です。
関数定義
torch.randn_like(input, dtype=None, device=None, requires_grad=False)
使用例
実例
import torch
# 入力テンソルを作成
input = torch.tensor([[1, 2], [3, 4]])
# 入力と同じ形状の標準正規分布に従うランダムテンソルを作成
x = torch.randn_like(input)
print(x)
# 入力テンソルを作成
input = torch.tensor([[1, 2], [3, 4]])
# 入力と同じ形状の標準正規分布に従うランダムテンソルを作成
x = torch.randn_like(input)
print(x)
その他の拡張