def _init_distribution(conditions): alpha = conditions["alpha"] return tfd.Zipf(power=alpha)
def _base_dist(self, alpha: TensorLike, *args, **kwargs): return tfd.Zipf(power=alpha, *args, **kwargs)
def _init_distribution(conditions): power = conditions["power"] return tfd.Zipf(power=power)
def _base_dist(self, alpha: TensorLike, *args, **kwargs): return tfd.Zipf(power=alpha, dtype=tf.float32, *args, **kwargs)