jax.experimental.pallas.triton.approx_tanh#
- jax.experimental.pallas.triton.approx_tanh(x)[源代码]#
逐元素的近似双曲正切函数：\(\mathrm{tanh}(x)\)。
请参阅 https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#floating-point-instructions-tanh。
逐元素的近似双曲正切函数：\(\mathrm{tanh}(x)\)。
请参阅 https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#floating-point-instructions-tanh。