[人工智能]PyTorch - torch.Tensor.repeat

PyTorch - torch.Tensor.repeat

flyfish

torch.Tensor.repeat
沿指定维度重复此张量

import torch
# 可以用行列来描述，更容易理解，如何重复[1, 2, 3]的问题
a = torch.tensor([1, 2, 3])
#tensor([1, 2, 3])
#如何重复[1, 2, 3]的问题
print(a.repeat(2)) # 2 * 原列数
#tensor([1, 2, 3, 1, 2, 3])

print(a.repeat(1,2)) #里面的数与a.repeat(2)相比没变，但是维度增加了。1* 原行数，2 * 原列数
#tensor([[1, 2, 3, 1, 2, 3]])

print(a.repeat(3, 2)) #3* 原行数，2 * 原列数
# tensor([[1, 2, 3, 1, 2, 3],
#         [1, 2, 3, 1, 2, 3],
#         [1, 2, 3, 1, 2, 3]])
print(a.repeat(4, 3, 2)) # 可以理解成4个  （3 * 原行数，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],
#          [1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3]],
#
#         [[1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3]],
#
#         [[1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3]]])

print(a.repeat(4, 3, 2).shape)
# torch.Size([4, 3, 6])