Open xtipacko opened 7 years ago
That would be amazing! Or alternatively (and arguably simpler than parsing a small expression DSL like einsum does) a tensordot
function that allows summing an arbitrary number of dimensions would provide all the same functionality. FWIW, I am working on autodiff of tensor valued functions for an ML application; in general, derivatives of matrix valued functions can't be expressed only in terms of matrix multiplications efficiently.
Good day! Could you please Implement something like numpy.einsum (for OpenCL and CUDA)?
In particular, I'm interested in row/column-wise outer product, like this, but einsum is very universal and useful thing...