我有一个(2,3,4)维的PyTorchTensor,我想让这个Tensor像(3,2,4)维。但是我想让(:,0,:)的值等于(0,:,:)。我尝试了torch. reform和torch.view,但是我没有得到预期的结果。我怎么能在PyTorch中做到这一点?
yuvru6vn1#
试试看:
import torch x = torch.randn(2, 0, 4) print(x.shape) x = x.permute(1, 0, 2) print(x.shape)
1条答案
按热度按时间yuvru6vn1#
试试看: