在通道上重复数组

Question

用于形状数组（示例，行，高度，通道数）。如何简单地将No替换为No？我搜索了np.repeat()，但应用失败。

import numpy as np
array = np.array([
                  [
                     [[0],[1]],
                     [[2],[3]],
                     [[4],[5]]
                  ],

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

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

                  [
                     [[0],[1]],
                     [[2],[3]],
                     [[4],[5]]
                  ]
                 ])
array.shape # (4, 3, 2, 1)

我想要一个形状数组（4、3、2、4）。频道应替换为一些培训示例。

Answer 1

您可以使用np.tile：

np.tile(array, (1, 1, 1, array.shape[0]))

或np.repeat：

np.repeat(array[:, :, :,], array.shape[0], axis=3)