python之将 Numpy 数组中的所有行对相乘

我有一个 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 中有一种更快的方法来完成这项工作。有人有什么想法吗？