更改pytorchTensor维度

mfuanj7w  于 2022-11-09  发布在  其他
关注(0)|答案(1)|浏览(159)

我有一个(2,3,4)维的PyTorchTensor,我想让这个Tensor像(3,2,4)维。但是我想让(:,0,:)的值等于(0,:,:)。我尝试了torch. reform和torch.view,但是我没有得到预期的结果。
我怎么能在PyTorch中做到这一点?

yuvru6vn

yuvru6vn1#

试试看:

import torch

x = torch.randn(2, 0, 4)
print(x.shape)

x = x.permute(1, 0, 2)
print(x.shape)

相关问题