jax.Array.dot

目录

jax.Array.dot#

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

计算两个数组的点积。

有关完整文档,请参阅 jax.numpy.dot()

参数:
  • self (Array)

  • b (ArrayLike)

  • precision (lax_internal.PrecisionLike)

  • preferred_element_type (DTypeLike | None)

返回类型:

Array