tensorplay._einsum.einsum

tensorplay._einsum.einsum(equation, *operands) → Tensor[source]

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]).