将二维tensor变为一维
时间: 2023-05-17 20:06:19 浏览: 846
PyTorch搭建一维线性回归模型(二)
tensor可以使用view函数来实现。例如,如果有一个形状为(3, 4)的二维tensor,可以使用以下代码将其变为一维tensor:
```python
import torch
# 创建一个形状为(3, 4)的二维tensor
x = torch.randn(3, 4)
# 将x变为一维tensor
y = x.view(-1)
print(y.shape) # 输出torch.Size([12])
```
其中,view函数的参数-1表示自动计算该维度的大小。在上面的例子中,由于原始tensor的大小为(3, 4),因此view函数将其变为了大小为12的一维tensor。
阅读全文