arnavkj1995 / vsg Goto Github PK

                          ```
                          class MySubclass(tfd.SomeDistribution):
                          
                            def __init__(self, param_a, param_b):
                              parameters = dict(locals())
                              # ... do subclass initialization ...
                              super(MySubclass, self).__init__(**base_class_params)
                              # Ensure that the subclass (not base class) parameters are stored.
                              self._parameters = parameters
                          
                            def _parameter_properties(self, dtype, num_classes=None):
                              return dict(
                                # Annotations may optionally specify properties, such as `event_ndims`,
                                # `default_constraining_bijector_fn`, `specifies_shape`, etc.; see
                                # the `ParameterProperties` documentation for details.
                                param_a=tfp.util.ParameterProperties(),
                                param_b=tfp.util.ParameterProperties())
                          ```