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