带广播的numpy阵列构造

Question

我希望将具有不同但广播兼容阵列的阵列堆叠在一起。给定7x5，7x1，1x5和1x1数组，我想

 a475 = mkarr([a75, a71, a15, a11])

其中a455将是一个4x7x5数组。 理想情况下我也希望能够做到

 a2275 = mkarr([[a75, a71], [a15, a11]])

获取2x2x7x5阵列。你知道吗

在numpy中表达这些操作最简洁的方式是什么？你知道吗

Answer 1

您可以使用^{}：

a475 = np.stack(np.broadcast_arrays(a75, a71, a15, a11))

请注意，这将创建原始数组的视图，因此不应占用任何额外的内存。你知道吗