将tensor1000维度的数据改成(1000,1)的数据
时间: 2024-02-22 07:00:09 浏览: 49
tensorflow 实现数据类型转换
您可以使用 PyTorch 中的 `unsqueeze` 函数将原来的 1000 维张量增加一个新的维度,使其变成形状为 (1000, 1) 的张量。示例如下:
```python
import torch
# 假设原来的数据为 tensor1000,形状为 (1000,)
tensor1000 = torch.randn(1000)
# 使用 unsqueeze 将数据变形成 (1000, 1) 的形状
tensor1000_reshaped = tensor1000.unsqueeze(1)
# 打印出新的张量的形状
print(tensor1000_reshaped.shape) # 输出 (1000, 1)
```
这样就可以将原来的 1000 维张量变成形状为 (1000, 1) 的张量了。
阅读全文