Skip to content

New class 'bij.Invert`, and `Bijector`s are now `nn.Module`'s
Compare
Choose a tag to compare
@stefanwebb stefanwebb released this 25 Apr 05:10
· 11 commits to main since this release
0.7
13cf226

This release add two new minor features.

A new class flowtorch.bijectors.Invert can be used to swap the forward and inverse operator of a Bijector. This is useful to turn, for example, Inverse Autoregressive Flow (IAF) into Masked Autoregressive Flow (MAF).

Bijector objects are now nn.Modules, which amongst other benefits allows easily saving and loading of state.

Assets 2