jax.lax.log#

jax.lax.log(x, accuracy=None)[来源]#

逐元素自然对数:\(\mathrm{log}(x)\)

此函数直接降到stablehlo.log 操作。

参数:
  • x (ArrayLike) – 输入数组。 必须具有浮点型或复数类型。

  • accuracy – 可选的 lax.Tolerancelax.AccuracyMode 对象,该对象根据请求的精度选择 op 的实现。如果实现无法满足请求的容差,编译器将返回错误。如果指定了模式,并且没有多个实现可用,将使用默认实现。

返回:

x 具有相同形状和 dtype 的数组,包含逐元素的自然对数。

返回类型:

数组

另请参阅