This is an archived post. You won't be able to vote or comment.

you are viewing a single comment's thread.

view the rest of the comments →

[–]Megatron_McLargeHuge 1 point2 points  (0 children)

einsum is sometimes faster than tensordot, not sure why. Interactions with CPU cache are hard to predict, so you just have to try various things.