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。