张 量 的 缩 并 运 算 张量的缩并运算

张量的缩并运算又称爱因斯坦求和规则:两个张量角标相同的维度进行求和


矩阵乘法就是缩并运算的一种。
矩阵就是二阶张量。


在python的numpy库中,提供了两个进行张量缩并的函数:
einsum :爱因斯坦求和规则,但是效率较低
tensordot:相比于einsum效率很高
假设有两个张量A(ijk)和 B(ijm)
einsum("ijk,ijm->km",A,B)
einsum("ijk,ijm->mk",A,B)