2 个矢量场的 Tensordot

Question

我想计算 2 个形状张量的元素张量积，这(1144,3)意味着如果我理解正确的话，我想计算沿第二个轴的张量点。

我希望我的结果是(1144,3,3).

我目前正在尝试使用 numpystensordot()函数来实现这一点，但我无法找出正确的轴来获得(1144,3,3).

Answer 1

您可以为此使用numpy.einsum 。In [30]: aOut[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.