jax.lax.mul

jax.lax.mul(x, y)

逐元素乘法：\(x \times y\)。

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

参数:

• x (ArrayLike) – 输入数组。必须具有匹配的数值数据类型。如果两者都不是标量，x 和 y 必须具有相同的维度数并且是广播兼容的。

• y (ArrayLike) – 输入数组。必须具有匹配的数值数据类型。如果两者都不是标量，x 和 y 必须具有相同的维度数并且是广播兼容的。

返回:

与 x 和 y 相同数据类型的数组，包含每对广播条目的乘积。

返回类型:

数组

另请参阅

• jax.numpy.multiply()：支持混合数据类型和秩的输入的 NumPy 风格乘法。