flâneur — a map of the web's best reading

Tim Rocktäschel

rockt.github.io · 2,151 words · saved by 1 readers

When talking to colleagues I realized that not everyone knows about einsum, my favorite function for developing deep learning models. This post is trying to change that once and for all! :) Einstein summation (einsum) is implemented in numpy, as well as deep learning libraries such as TensorFlow and, thanks to Thomas Viehmann, recently also PyTorch. For background reading on einsum, I recommend the excellent blog posts by Olexa Bilaniuk and Alex Riley. While their posts discuss einsum in the context of numpy, I am going to illustrate how einsum is extremely useful for writing elegant PyTorch/TensorFlow models.1

Tim Rocktäschel Tim Rocktäschel Artificial General Intelligence, Open-endedness, Self-Improvement, Reinforcement Learning Spatial Lorem ipsum dolor sit amet nullam consequat interdum vivamus donce sed libero. Get Started --> – Tim Rocktäschel, 30/04/2018 – updated 02/05/2018 When talking to colleagues I realized that not everyone knows about einsum , my favorite function for developing deep learning models. This post is trying to change that once and for all! :) Einstein summation (einsum) is implemented in numpy , as well as deep learning libraries such as TensorFlow and, thanks

Explore this link on the map →

related reading