lua - Torch - 像numpy重复一样重复张量

Question

我试图以两种方式在火炬中重复张量。例如重复张量 {1,2,3,4}双向3次屈服；

{1,2,3,4,1,2,3,4,1,2,3,4}
{1,1,1,2,2,2,3,3,3,4,4,4}

有一个内置的 torch:repeatTensor 函数，它将生成两者中的第一个(如 numpy.tile() )，但我找不到后者(如 numpy.repeat() )。我确定我可以在第一个上调用 sort 来给第二个，但我认为这对于更大的数组来说可能计算成本高吗？

谢谢。

Answer 1

尝试 torch.repeat_interleave() 方法:https://pytorch.org/docs/stable/torch.html#torch.repeat_interleave

>>> x = torch.tensor([1, 2, 3])
>>> x.repeat_interleave(2)
tensor([1, 1, 2, 2, 3, 3])