如何将一个数组复制/重复 N 次到一个新数组中？

Question

我有：

 test = np.random.randn(40,40,3)

我想做：

 result = Repeat(test, 10)

因此 result 包含数组 test 重复10次，形状为：

 (10, 40, 40, 3)

因此，创建一个带有新轴的张量以容纳 test 的 10 个副本。我也想尽可能高效地做到这一点。我怎样才能用 Numpy 做到这一点？

原文由 JDS 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

可以将 np.repeat 方法与 np.newaxis 一起使用：

 import numpy as np

test = np.random.randn(40,40,3)
result = np.repeat(test[np.newaxis,...], 10, axis=0)
print(result.shape)
>> (10, 40, 40, 3)

原文由 OddNorg 发布，翻译遵循 CC BY-SA 4.0 许可协议