search

april-tools / cirkit

a python framework to build, learn and reason about probabilistic circuits and tensor networks
https://cirkit-docs.readthedocs.io/en/latest/
GNU General Public License v3.0
71 stars 1 forks source link

Introduce `Reprameterization` class #140

Closed lkct closed 11 months ago

lkct commented 11 months ago

Closes #136

  1. Fixed the wrong way of masking and the wrong use of nan_to_num.
  2. Changed reparam from functions to classes. 2.1. Only leaves (base reparams) are implemented. 2.2. TODO: compose reparams, e.g. product of two.
  3. Provided full shaped fold_mask to Layers (then to Reprameterization). There's no need to unsqueeze in layers.
loreloc commented 11 months ago

Thanks!