Python PyTorch tile用法及代码示例

本文简要介绍python语言中 torch.tile 的用法。

用法:

torch.tile(input, dims) → Tensor

参数：

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

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

通过重复 input 的元素构造一个张量。 dims 参数指定每个维度中的重复次数。

如果 dims 指定的维数少于 input 的维数，那么将在 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]])