當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python SciPy linalg.khatri_rao用法及代碼示例

本文簡要介紹 python 語言中 scipy.linalg.khatri_rao 的用法。

用法:

scipy.linalg.khatri_rao(a, b)#

Khatri-rao產品

兩個矩陣的按列 Kronecker 積

參數

a (n, k) 數組

輸入數組

b (m, k) 數組

輸入數組

返回

c: (n*m, k) 數組

Khatri-rao a 和 b 的乘積。

注意

Khatri-Rao 乘積的數學定義是:

這是 A 和 B 每一列的克羅內克積,例如:

c = np.vstack([np.kron(a[:, k], b[:, k]) for k in range(b.shape[1])]).T

例子

>>> import numpy as np
>>> from scipy import linalg
>>> a = np.array([[1, 2, 3], [4, 5, 6]])
>>> b = np.array([[3, 4, 5], [6, 7, 8], [2, 3, 9]])
>>> linalg.khatri_rao(a, b)
array([[ 3,  8, 15],
       [ 6, 14, 24],
       [ 2,  6, 27],
       [12, 20, 30],
       [24, 35, 48],
       [ 8, 15, 54]])

相關用法


注:本文由純淨天空篩選整理自scipy.org大神的英文原創作品 scipy.linalg.khatri_rao。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。