einsum wrapper?

Open kevin-w-li opened this issue 8 years ago • 2 comments

Is there a way we can use einsum from tensorflow? https://www.tensorflow.org/api_docs/python/tf/einsum

Oct 26 '17 11:10 kevin-w-li

Not at the moment. It would be great to have, but the logic would have to be ported from Python to Julia.

Oct 26 '17 13:10 malmaud

Einsum.jl would almost work. For the static case, where all sizes are known in advance.

What is required to make that work is to switch size to be opportunistically static (get_shape when possible, rather than always shape), and make setindex! work.

It wouldn't be had to write that static case as a one off using get_shape and scatter. It wouldn't be too hard to do the dynamic case by adding in @tf while.

Oct 27 '17 00:10 oxinabox