batch shape and event shape - bdatko/numpyro_play GitHub Wiki Required reading Reasoning about Shapes and Probability Distributions The amazing blog post by Eric J. Ma, cleared up so much confusion Understanding TensorFlow Distributions Shapes More resources Shape-semantics Shapes in distribution Tensor shapes in Pyro PyTorch layer dimensions: what size and why?