PyTorch Developer Podcast
PyTorch Developer Podcast
Edward Yang, Team PyTorch
Batching
13 minutes Posted Aug 18, 2021 at 1:00 pm.
0:00
13:37
Download MP3
Show notes

PyTorch operates on its input data in a batched manner, typically processing multiple batches of an input at once (rather than once at a time, as would be the case in typical programming). In this podcast, we talk a little about the implications of batching operations in this way, and then also about how PyTorch's API is structured for batching (hint: poorly) and how Numpy introduced a concept of ufunc/gufuncs to standardize over broadcasting and batching behavior. There is some overlap between this podcast and previous podcasts about TensorIterator and vmap; you may also be interested in those episodes.

Further reading.