pytorch-函数使用- repeat
x = torch.randn(4, 1)
xx = x.repeat(1, 5)
output:
x:tensor([[ 0.6720],
[-0.2683],
[ 1.0877],
[-1.8444]])
xx:
tensor([[ 0.6720, 0.6720, 0.6720, 0.6720, 0.6720],
[-0.2683, -0.2683, -0.2683, -0.2683, -0.2683],
[ 1.0877, 1.0877, 1.0877, 1.0877, 1.0877],
[-1.8444, -1.8444, -1.8444, -1.8444, -1.8444]])
x的shape为[4,1],xx的shape为[4,5]