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

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay._composite_funcs.quantile

Compute quantiles through the dispatcher composite implementation.

tensorplay._shape_funcs.block_diag

Builds a block diagonal matrix from the given blocks.

tensorplay._einsum.einsum

Help improve this page

On this page