快捷方式

torch.tile

torch.tile(input, dims) Tensor

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

如果 dims 指定的维度少于 input 的维度,则将 1 预先添加到 dims 直到所有维度都被指定。 例如,如果 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]])

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源