tfp.substrates.jax.math.smootherstep

Computes a sigmoid-like interpolation function on the unit-interval.

tfp.substrates.jax.math.smootherstep(
    x, name=None
)

Equivalent to:

x = tf.clip_by_value(x, clip_value_min=0., clip_value_max=1.)
y = x**3. * (6. * x**2. - 15. * x + 10.)

For more details see [Wikipedia][1].

x float Tensor.
name Python str name prefixed to Ops created by this function. Default value: None (i.e., 'smootherstep').

smootherstep float Tensor with the same shape and dtype as x, representing the value of the smootherstep function.

References

[1]: "Smoothstep." Wikipedia. https://en.wikipedia.org/wiki/Smoothstep#Variations