将 Numpy 数组中的所有行相乘

Question

我有一个 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 中有一种更快的方法来做到这一点。有没有人有任何想法？

Answer 1

您可以使用np.einsum：np.einsum('ik,jk->ijk', le_input, le_input)# array([[[0, 0, 1],#         [0, 0, 0]],#        [[0, 0, 0],#         [0, 1, 0]]])或者创建一个新轴并使用数组的广播属性来计算第一维上的外积：le_input[:,None] * le_input# array([[[0, 0, 1],#         [0, 0, 0]],#        [[0, 0, 0],#         [0, 1, 0]]])