Nice! I remember trying to implement Hinton's capsules layer in tensorflow [1], and stumbling upon Einstein notation syntax. Took me a while to figure it out.

[1] https://github.com/michaelklachko/CapsNet/blob/master/capsne...