jax.lax.cos

jax.lax.cos(x, accuracy=None)

逐元素余弦: \(\mathrm{cos}(x)\).

对于浮点输入，此函数直接降级为 stablehlo.cosine 操作。对于复杂输入，它降低为实现复杂余弦的 HLO 操作序列。

参数:

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

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

返回:

与 x 具有相同形状和 dtype 的数组，包含逐元素余弦。

返回类型:

数组

另请参阅

• jax.lax.sin(): 逐元素正弦。

• jax.lax.tan()：逐元素正切。

• jax.lax.acos()：逐元素反余弦。