tfp.math.log1psquare

tfp.math.log1psquare(x)

A numerically stable implementation of log(1 + x**2).