如何在轴= 1内连接张量？

Question

我有一个形状张量(2,2,2,2)：tensor([[[[   5.,    5.],          [   5.,    5.]],         [[  10.,   10.],          [  10.,   10.]]],        [[[ 100.,  100.],          [ 100.,  100.]],         [[1000., 1000.],          [1000., 1000.]]]], device='cuda:0')我想对其进行转换，以使沿 axis=1 的张量重复 3 次。在应用之后.view(-1)，我得到一个 1D 结果张量：tensor([   5.,    5.,    5.,    5.,   5.,    5.,    5.,    5.,   5.,    5.,    5.,    5.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,  100.,  100.,  100.,  100.,  100.,  100., 100.,  100.,  100.,  100., 100.,  100.,  100.,  100.,  100.,  100., 1000., 1000., 1000., 1000.  1000., 1000., 1000., 1000. 1000., 1000., 1000., 1000. 1000., 1000., 1000., 1000.], device='cuda:0')这个怎么做？

Answer 1

用于torch.repeat_interleave重复张量的元素。

t.repeat_interleave(repeats=3, dim=1).view(-1)

tensor([   5.,    5.,    5.,    5.,    5.,    5.,    5.,    5.,    5.,    5.,

           5.,    5.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,

          10.,   10.,   10.,   10.,  100.,  100.,  100.,  100.,  100.,  100.,

         100.,  100.,  100.,  100.,  100.,  100., 1000., 1000., 1000., 1000.,

        1000., 1000., 1000., 1000., 1000., 1000., 1000., 1000.])

Answer 2

尝试这个：

final_features = feature_combined.view(1, -1, feature_combined.size(-1))

                                 .repeat(1, 1, 3).view(1, -1).squeeze(0)

我假设您输入的形状为 的张量(2, 2, 2, 2)在feature_combined. final_features结果的形状([48])如您所愿。

print(feature_combined)

>>tensor([   5.,    5.,    5.,    5.,    5.,    5.,    5.,    5.,    5.,    5.,

       5.,    5.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,   10.,

      10.,   10.,   10.,   10.,  100.,  100.,  100.,  100.,  100.,  100.,

     100.,  100.,  100.,  100.,  100.,  100., 1000., 1000., 1000., 1000.,

    1000., 1000., 1000., 1000., 1000., 1000., 1000., 1000.])