Resources - batch shape and event shape explained

Hi all,

If anyone is having some trouble understanding the semantics behind batch_shape and event_shape or want to know more after reading Tensor shapes in Pyro I would recommend reading the excellent post Reasoning about Shapes and Probability Distributions by Eric J. Ma.

Another source would be TensorFlow’s probability post Understanding TensorFlow Distributions Shapes (where I found Eric Ma’s post). Just be careful because not everything is applicable to Pyro and NumPyro since it’s TensorFlow.

Hope it helps. =)


@bdatko Thanks for sharing! It really helped me understanding the semantics.

1 Like