我想計(jì)算 2 個(gè)形狀張量的元素張量積，這(1144,3)意味著如果我理解正確的話，我想計(jì)算沿第二個(gè)軸的張量點(diǎn)。我希望我的結(jié)果是(1144,3,3).我目前正在嘗試使用 numpystensordot()函數(shù)來(lái)實(shí)現(xiàn)這一點(diǎn)，但我無(wú)法找出正確的軸來(lái)獲得(1144,3,3).
1 回答
滄海一幻覺(jué)
TA貢獻(xiàn)1824條經(jīng)驗(yàn) 獲得超5個(gè)贊
您可以為此使用numpy.einsum 。
In [30]: a
Out[30]:
array([[0, 1, 2],
[3, 4, 5]])
In [31]: np.einsum('ij,ik->ijk', a, a)
Out[31]:
array([[[ 0, 0, 0],
[ 0, 1, 2],
[ 0, 2, 4]],
[[ 9, 12, 15],
[12, 16, 20],
[15, 20, 25]]])
由于numpy.tensordot僅支持 2 個(gè)元素，axes這意味著無(wú)法模仿
->...- 樣的行為。所以我看不出如何用numpy.tensordot.
添加回答
舉報(bào)
0/150
提交
取消