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 函数。
示例
>>> 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]])