PyTorch torch.rand 関数
Pytorch torch リファレンスマニュアルtorch.randPyTorch で、一様分布 [0, 1) からランダムテンソルを作成するための関数です。
これは深層学習において、初期化やランダムマスクの生成などのシナリオでよく使用されます。
関数定義
torch.rand(*size, dtype=None, device=None, requires_grad=False)
引数:
*size(int): テンソルの形状。dtype(torch.dtype, オプション): データ型。device(torch.device, オプション): デバイス。requires_grad(bool, オプション): 勾配を計算する必要があるかどうか。
戻り値:
torch.Tensor: 範囲 [0, 1) の乱数を含むテンソルを返します。
使用例
例 1: ランダムテンソルの作成
実例
import torch
# 3x4 のランダムテンソルを作成
x = torch.rand(3, 4)
print(x)
print("範囲: [{:.4f}, {:.4f})".format(x.min().item(), x.max().item()))
# 3x4 のランダムテンソルを作成
x = torch.rand(3, 4)
print(x)
print("範囲: [{:.4f}, {:.4f})".format(x.min().item(), x.max().item()))
出力結果：
tensor([[0.6272, 0.1802, 0.8336, 0.3900],
[0.2188, 0.6593, 0.3598, 0.4021],
[0.9247, 0.0144, 0.2587, 0.6634]])
范围: [0.0144, 0.9247)
例 2: ランダムマスクの作成
実例
import torch
# 50% の確率のランダムマスクを作成
mask = torch.rand(10) > 0.5
print("マスク:", mask)
print("True の数:", mask.sum().item())
# 50% の確率のランダムマスクを作成
mask = torch.rand(10) > 0.5
print("マスク:", mask)
print("True の数:", mask.sum().item())
その他の拡張