2 个矢量场的 Tensordot

Question

我想计算 2 个形状张量的元素张量积，这(1144,3)意味着如果我理解正确的话，我想计算沿第二个轴的张量点。我希望我的结果是(1144,3,3).我目前正在尝试使用 numpystensordot()函数来实现这一点，但我无法找出正确的轴来获得(1144,3,3).

Answer 1

您可以为此使用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 个元素，axes这意味着无法模仿

->...- 样的行为。所以我看不出如何用numpy.tensordot.