pytorch入门（十）：repeat

import numpy as np

a = np.arange(6).reshape(2,3)
b = a.repeat(2,0)
print(b)
b = a.repeat(2,1)
print(b)

运行结果

[[0 1 2]
 [0 1 2]
 [3 4 5]
 [3 4 5]]
[[0 0 1 1 2 2]
 [3 3 4 4 5 5]]

类似的代码，把np换成torch，方向也变成1和2

import torch
a = torch.arange(6).reshape(2,3)
b = a.repeat(2,1)
print(b)
b = a.repeat(2,2)
print(b)

运行结果

tensor([[0, 1, 2],
        [3, 4, 5],
        [0, 1, 2],
        [3, 4, 5]])
tensor([[0, 1, 2, 0, 1, 2],
        [3, 4, 5, 3, 4, 5],
        [0, 1, 2, 0, 1, 2],
        [3, 4, 5, 3, 4, 5]])

【上一篇：处理异或问题】