pytorch 生成随机向量
时间: 2023-10-06 22:11:05 浏览: 245
Pytorch生成随机数Tensor的方法汇总
在PyTorch中,可以使用torch.randn函数生成一个具有随机值的张量,该张量的值服从标准正态分布(均值为0,标准差为1)。
以下是生成随机向量的示例代码:
```python
import torch
# 生成一个具有随机值的一维向量,长度为5
random_vector = torch.randn(5)
print(random_vector)
```
输出结果可能类似于:
```
tensor([-0.3152, 0.4239, 0.6795, -1.1025, -0.3816])
```
在上述示例中,torch.randn(5)会生成一个长度为5的一维向量,其中的值是从标准正态分布中随机抽样得到的。
如果需要生成多维的随机向量,可以根据需要设置相应的维度大小。例如,使用torch.randn(3, 4)可以生成一个3行4列的二维随机向量。
阅读全文