矩陣數(shù)學(xué)的Pythonnumpy函數(shù)

這是你想要的嗎？

>>> a
array([[1, 2],
       [2, 3],
       [3, 4],
       [5, 6]])

>>> b
array([[ 2,  4],
       [ 6,  8],
       [10, 11]])

>>> a[:,None,:]*b
array([[[ 2,  8],
        [ 6, 16],
        [10, 22]],

       [[ 4, 12],
        [12, 24],
        [20, 33]],

       [[ 6, 16],
        [18, 32],
        [30, 44]],

       [[10, 24],
        [30, 48],
        [50, 66]]])

>>> _.shape
(4, 3, 2)

或者，如果形狀不正確，可以對(duì)其進(jìn)行重塑：

>>> (a[:,None,:]*b).reshape((a.shape[0]*b.shape[0], 2))
array([[ 2,  8],
       [ 6, 16],
       [10, 22],
       [ 4, 12],
       [12, 24],
       [20, 33],
       [ 6, 16],
       [18, 32],
       [30, 44],
       [10, 24],
       [30, 48],
       [50, 66]])