einopsEasy Playground

Flexible tensor operations with readable Einstein notation

Getting started with einopsRun locally
Install
pip install einops
Python CodeRun locally
Expected Output
# Expected output shown below
# (Run locally with: einops)

einops is a third-party package. Flexible tensor operations with readable Einstein notation. Install with: pip install einops

Challenge

Try modifying the code above to explore different behaviors. Can you extend the example to handle a new use case?