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