时间戳

首页 > 解决方案 > 按自定义顺序重塑张量（PyTorch）

python - 按自定义顺序重塑张量（PyTorch）

问题描述

我有以下张量

t = torch.tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14 ,15, 16, 17])

我想通过以下方式重塑它：

t_reshape = torch.tensor([[0, 1, 2, 6, 7, 8, 12, 13, 14], 
                          [3, 4, 5, 9, 10, 11, 15, 16, 17]])

有没有办法以这种方式有效地重塑张量？

标签： pythonpytorchreshapetensor

解决方案

您可以通过重新整形、转置和重新整形来实现这一点：

>>> t.reshape(3,2,-1).transpose(0,1).reshape(2,-1)
tensor([[ 0,  1,  2,  6,  7,  8, 12, 13, 14],
        [ 3,  4,  5,  9, 10, 11, 15, 16, 17]])

推荐阅读