danielward27 / flowjax

https://danielward27.github.io/flowjax/
MIT License
82 stars 10 forks source link

Reshape #138

Closed danielward27 closed 6 months ago

danielward27 commented 6 months ago

Add a Reshape bijection. One use for this is for defining scalar flows, even if the bijections force flow.ndim==1 on construction.