search

pyro-ppl / funsor

Functional tensors for probabilistic programming
https://funsor.pyro.ai
Apache License 2.0
236 stars 20 forks source link

Use new ops to reduce multiple dims in eager_reduce #505

Closed ordabayevy closed 3 years ago

ordabayevy commented 3 years ago

This simplifies eager_reduce by using new ops (#482) that can reduce over multiple dims.