jax.Array.dot#

抽象 Array.dot(b, *, precision=None, preferred_element_type=None)[源代码]#

计算两个数组的点积。

请参阅 jax.numpy.dot() 获取完整文档。

参数:
  • self (Array)

  • b (ArrayLike)

  • precision (lax.PrecisionLike)

  • preferred_element_type (DTypeLike | None)

返回类型:

Array