tensor:10变成tensor[10,1]
时间: 2023-11-21 16:05:05 浏览: 141
pytorch 改变tensor尺寸的实现
可以使用 PyTorch 中的 `unsqueeze` 方法将一个数值为 10 的标量转换为形状为 `[10,1]` 的张量。具体代码如下:
```python
import torch
x = torch.tensor(10)
x = x.unsqueeze(0).unsqueeze(1)
print(x)
```
输出:
```
tensor([[10]])
```
阅读全文