<分区>
我有一个 MxN Numpy 数组。我想获取数组的每一行并将其按元素乘以数组的每一行,从而得到一个 MxMxN numpy 乘积数组。
le_input = np.array([
[0, 0, 1],
[0, 1, 0]
])
le_expected_output = np.array([
[
[0, 0, 1],
[0, 0, 0]
],
[
[0, 0, 0],
[0, 1, 0]
]
])
我当然可以使用 for 循环来完成这项工作,我已经尝试过了,但我假设在 Numpy 中有一种更快的方法来完成这项工作。有人有什么想法吗?