tensorplay._einsum API

Sums the product of the elements of the input operands along dimensions specified using a notation based on the Einstein summation convention, e.g. tp.einsum("ij,jk->ik", A, B) computes a matrix multiplication.

tp.einsum(A, [..., 0, 1], B, [..., 1, 2], [..., 0, 2]).