如何将torch.randn(3)转换成1行3列的向量
时间: 2023-06-26 15:08:32 浏览: 87
pytorch:torch.mm()和torch.matmul()的使用
可以使用torch.reshape()函数将torch.randn(3)转换成1行3列的向量。示例如下:
```
import torch
x = torch.randn(3)
print(x)
# 将x转换成1行3列的向量
x = torch.reshape(x, (1, 3))
print(x)
```
输出结果为:
```
tensor([-0.2717, 0.3071, -0.3091])
tensor([[-0.2717, 0.3071, -0.3091]])
```
阅读全文