torch.tile

torch.tile(input, dims) → Tensor

通过重复 input 的元素来构建张量。dims 参数指定了每个维度上的重复次数。

如果 dims 指定的维度少于 input 的维度，则会在 dims 前面填充 1，直到指定所有维度。例如，如果 input 的形状是 (8, 6, 4, 2)，而 dims 是 (2, 2)，则 dims 会被视为 (1, 1, 2, 2)。

类似地，如果 input 的维度少于 dims 指定的维度，则 input 会被视为在维度零处被 unsqueeze（扩充维度），直到其维度数量与 dims 指定的一致。例如，如果 input 的形状是 (4, 2)，而 dims 是 (3, 3, 2, 2)，则 input 会被视为具有 (1, 1, 4, 2) 的形状。

注意

此函数类似于 NumPy 的 tile 函数。

参数

• input (Tensor) – 要重复其元素的张量。

• dims (tuple) – 每个维度上的重复次数。

示例

>>> x = torch.tensor([1, 2, 3])
>>> x.tile((2,))
tensor([1, 2, 3, 1, 2, 3])
>>> y = torch.tensor([[1, 2], [3, 4]])
>>> torch.tile(y, (2, 2))
tensor([[1, 2, 1, 2],
        [3, 4, 3, 4],
        [1, 2, 1, 2],
        [3, 4, 3, 4]])