tensor reshape 成一个三维向量
时间: 2023-06-07 14:09:47 浏览: 122
pytorch改变tensor维度 - Python-PyTorch reshape Tensor维度
好的,可以回答该问题。对一个Tensor进行reshape可以使用reshape()函数,比如对一个2维的Tensor进行reshape为3维的Tensor,可以使用以下代码:
```
import torch
# 假设现在有一个2*4的tensor
t = torch.randint(0, 10, size=(2, 4))
print(t) # 输出原始tensor
# reshape成3维的Tensor,第一维度为2,后面两个维度为2
t = t.reshape(2, 2, 2)
print(t) # 输出reshape完毕的tensor
```
这个例子将一个2\*4的Tensor reshape成了一个2\*2\*2的Tensor。
阅读全文