pytorch randn创建的矩阵
时间: 2023-10-10 08:12:02 浏览: 87
pytorch 转换矩阵的维数位置方法
`torch.randn()` 函数可以用来创建一个指定大小的张量,张量中的元素是从标准正态分布(均值为0,方差为1)中随机抽样得到的。下面是一个创建大小为 (3, 4) 的随机张量的例子:
```python
import torch
rand_tensor = torch.randn(3, 4)
print(rand_tensor)
```
输出结果如下所示:
```
tensor([[-1.0311, -0.1327, -0.6517, -0.1878],
[ 1.2776, 0.3116, 0.9398, 1.8699],
[-0.0627, 0.0658, -2.3764, 0.9018]])
```
注意,`torch.randn()` 函数生成的每个元素都是独立的随机数。如果需要生成一个固定种子的随机张量,可以使用 `torch.manual_seed()` 函数。例如:
```python
import torch
torch.manual_seed(42)
rand_tensor = torch.randn(3, 4)
print(rand_tensor)
```
输出结果:
```
tensor([[-0.2347, -0.4102, 0.7610, 0.1217],
[ 0.4439, -0.2553, 1.0105, -0.0293],
[-0.1098, -0.4414, -0.1094, 0.2412]])
```
阅读全文