Python PyTorch Tensor.repeat用法及代碼示例

本文簡要介紹python語言中 torch.Tensor.repeat 的用法。

用法:

Tensor.repeat(*sizes) → Tensor

參數：

sizes(torch.Size或者詮釋...) -沿每個維度重複此張量的次數

沿指定維度重複此張量。

與 expand() 不同，此函數複製張量的數據。

警告

repeat() 的行為與 numpy.repeat 不同，但與 numpy.tile 更相似。對於類似於 numpy.repeat 的運算符，請參閱 torch.repeat_interleave() 。

例子：

>>> x = torch.tensor([1, 2, 3])
>>> x.repeat(4, 2)
tensor([[ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3]])
>>> x.repeat(4, 2, 1).size()
torch.Size([4, 2, 3])