jax.lax.dot

目录

jax.lax.dot#

jax.lax.dot(lhs, rhs, precision=None, preferred_element_type=None)[源代码][源代码]#

向量/向量、矩阵/向量和矩阵/矩阵乘法。

封装了 XLA 的 Dot 操作符。

更多关于一般收缩的信息，请参见 jax.lax.dot_general() 操作符。

参数:

  • lhs (Array) – 一个维度为1或2的数组。

  • rhs (Array) – 一个维度为1或2的数组。

  • precision (PrecisionLike) – 可选。可以是 None，这意味着后端的默认精度，一个 Precision 枚举值（Precision.DEFAULT、Precision.HIGH 或 Precision.HIGHEST），或者是一个包含两个 Precision 枚举的元组，指示 lhs` 和 rhs 的精度。

  • preferred_element_type (DTypeLike | None) – 可选。可以是 None，这意味着使用输入类型的默认累积类型，或者是一个数据类型，指示将结果累积到并返回该数据类型的结果。

返回:

包含产品的数组。

返回类型:

Array

目录