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