torch.tile

torch.tile(input, dims) → Tensor

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

如果 dims 指定的维度少于 input ，则将 1 们添加到 dims 中，直到所有维度都被指定。例如，如果 input 的形状为(8, 6, 4, 2)，而 dims 为(2, 2)，则 dims 被视为(1, 1, 2, 2)。

类似地，如果 input 的维度少于 dims 指定的，则 input 被视为在维度零上未压缩，直到它具有 dims 指定的维度数。例如，如果 input 的形状为(4, 2)，而 dims 为(3, 3, 2, 2)，则 input 被视为具有形状(1, 1, 4, 2)。

注意

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

参数:

• 输入（张量）- 要重复其元素的张量。

• 每维度的重复次数（元组）。

示例：

>>> 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]])