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