Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfp.experimental.substrates.jax.math.log_cosh

View source on GitHub

Compute log(cosh(x)) in a numerically stable way.

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

Args:

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

Returns:

  • log_cosh: log_cosh(x).