I am new to TensorFlow and I am trying to implement a model that requires an inverse cdf for a Beta distribution.

I implemented a similar one for a normal distribution using tfp.distributions.Normal.quantile() which worked well.

However, when I try it for the Beta distribution I get the following error:

NotImplementedError: quantile is not implemented: Beta

I tried a work around using the scipy function: scipy.stats.beta.ppf. However, I am encountering a lot of OperatorNotAllowedInGraphError type errors when I try and incorporate it into a TensorFlow function.

Any help with either of the above issues would be greatly appreciated.