Enable nsc lib to be in functional form

Photrek / Nonlinear-Statistical-Coupling

Apache License 2.0

5 stars 1 forks source link

Enable nsc lib to be in functional form #30

Open Kevin-Chen0 opened 3 years ago

Kevin-Chen0 commented 3 years ago

Instantiation distribution classes do not require passing in hard loc and scale 'tensors' right away, allowing for delayed execution. This allows for speedier runtime during the execution of the model. For example, @tf.function does not have to be commented out when integrated nsc into a TF-based model.

jkclem commented 3 years ago

I believe this has been implemented now, although we are using special versions of the functions to avoid having to instantiate MultivariateCoupledNormal objects.