torch 生成随机数
时间: 2023-10-30 08:05:53 浏览: 169
torch_spline_conv-1.2.2-cp311-cp311-macosx_10_9_universal2whl
可以使用PyTorch中的torch.randn()函数来生成随机数。这个函数可以生成一个具有标准正态分布(均值为0,方差为1)的随机数张量。例如,要生成一个形状为(3, 3)的随机数张量,可以使用以下代码:
```python
import torch
random_tensor = torch.randn(3, 3)
print(random_tensor)
```
这将输出一个形状为(3, 3)的随机数张量。如果你想生成具有不同均值和方差的随机数张量,可以使用torch.randn_like()函数并指定期望的均值和方差。例如:
```python
mean = 5
std = 2
random_tensor = torch.randn_like(random_tensor) * std + mean
print(random_tensor)
```
这将生成一个具有均值为5,标准差为2的随机数张量。希望这可以帮到你!
阅读全文