Einstein sum feature is awesome!

Deep learners will find torch.einsum() helpful in multiple ways. Essentially, you can use it to carry out any sort of operation on two tensors, thus helping circumvent loops. See https://rockt.github.io/2018/04/30/einsum for an intro and a few examples.

1 Like

@tom implemented this ā€“ thank you @tom!

1 Like

Well done @tom! :+1:
It looks like a very nice feature.

Heya, thanks for the flowers. Iā€™m glad it is of use.

1 Like