pytorchで正規分布を作成する方法

Question

Pytorchでランダムな正規分布を作成したいのですが、平均値と標準値はそれぞれ4、0.5です。そのためのAPIが見つかりませんでした。誰か知ってる？どうもありがとう。

Answer 1

何を生成したいかによります。

標準正規分布を生成するために-

_torch.randn()
_

すべての分布（正規、ポアソン、ユニフォームなど）については、torch.distributions.Normal()またはtorch.distribution.Uniform()を使用します。これらすべてのメソッドの詳細は、こちらで確認できます- https://pytorch.org/docs/stable/distributions.html#normal

これらのメソッドを定義したら、.sampleメソッドを使用してインスタンスの数を生成できます。また、分布パラメーターがバッチ処理されている場合は、sample_shape形状のサンプルまたはsample_shape形状のサンプルのバッチを生成できます。

Answer 2

単純なオプションは、基本モジュールのrandn関数を使用することです。標準のガウス分布からランダムなサンプルを作成します。平均と標準偏差を変更するには、加算と乗算を使用するだけです。以下に、リクエストされたディストリビューションからサイズ5のサンプルを作成します。

import torch
torch.randn(5) * 0.5 + 4 # tensor([4.1029, 4.5351, 2.8797, 3.1883, 4.3868])