[Python] einsum is all you need

매트릭스 계산을 나름 직관적으로 표현할 수 있지만 헷갈림

einsum is all you need - einsum 용법 정리

https://rockt.github.io/2018/04/30/einsum

Tim Rocktäschel

 

pytorch einsum 함수 정리

https://pytorch.org/docs/stable/generated/torch.einsum.html#torch.einsum

torch.einsum — PyTorch 1.10 documentation