PyTorch torch.randint_like 関数
Pytorch torch リファレンスマニュアルtorch.randint_likeこれは PyTorch において、入力テンソルと同じ形状の整数ランダムテンソルを作成するための関数です。
関数定義
torch.randint_like(input, low=0, high, dtype=None, device=None, requires_grad=False)
使用例
例
import torch
# 入力テンソルを作成する
input = torch.tensor([[1, 2], [3, 4]])
# 入力と同じ形状の整数ランダムテンソルを作成する。範囲 [0, 10)
x = torch.randint_like(input, 10)
print(x)
# 入力テンソルを作成する
input = torch.tensor([[1, 2], [3, 4]])
# 入力と同じ形状の整数ランダムテンソルを作成する。範囲 [0, 10)
x = torch.randint_like(input, 10)
print(x)
その他の拡張