将两个数组聚合成一个具有给定形状的数组的正确方法

Question

有两个形状为[1, 255, 255, 1]的数组，我想将它们聚合成[1,255, 255,2]。在 numpy 中正确的做法是什么？

我正在尝试使用

import numpy as np
A = np.append(A1, A2)

但是好像不太对

Answer 1

你快到了：

A = np.append(A1, A2).reshape(1, 255, 255, 2)

Answer 2

再难一点！

AA= np.concatenate((A1,A2), axis=-1)

由于 2 个数组在维度数上匹配，并且您想在最后一个轴 (-1) 上连接它们，并且其他维度匹配，只需使用 concatenate.

其实大部分的数组连接任务都可以用concatenate来完成。有些可能需要添加尺寸。但是这里的东西已经匹配了。