torch.normal()
torch.normal(means, std, out=None)
返回一個張量,包含從給定參數means
,std
的離散正態分佈中抽取隨機數。 均值means
是一個張量,包含每個輸出元素相關的正態分佈的均值。 std
是一個張量,包含每個輸出元素相關的正態分佈的標準差。 均值和標準差的形狀不須匹配,但每個張量的元素個數須相同。
參數:
- means (Tensor) – 均值
- std (Tensor) – 標準差
- out (Tensor) – 可選的輸出張量
torch.normal(means=torch.arange(1, 11), std=torch.arange(1, 0, -0.1))
1.5104
1.6955
2.4895
4.9185
4.9895
6.9155
7.3683
8.1836
8.7164
9.8916
[torch.FloatTensor of size 10]
torch.normal(mean=0.0, std, out=None)
與上面函數類似,所有抽取的樣本共享均值。
參數:
- means (Tensor,optional) – 所有分佈均值
- std (Tensor) – 每個元素的標準差
- out (Tensor) – 可選的輸出張量
例子:
>>> torch.normal(mean=0.5, std=torch.arange(1, 6))
0.5723
0.0871
-0.3783
-2.5689
10.7893
[torch.FloatTensor of size 5]
torch.normal(means, std=1.0, out=None)
與上面函數類似,所有抽取的樣本共享標準差。
參數:
- means (Tensor) – 每個元素的均值
- std (float, optional) – 所有分佈的標準差
- out (Tensor) – 可選的輸出張量
例子:
>>> torch.normal(means=torch.arange(1, 6))
1.1681
2.8884
3.7718
2.5616
4.2500
[torch.FloatTensor of size 5]