tensorplay.linalg.tensorsolve

tensorplay.linalg.tensorsolve(A, B, dims=None) → Tensor[source]

Solves the tensor equation A X = B after flattening the contracted dimensions into a square matrix. dims identifies dimensions of A that should be moved to the trailing side before the flattening step.