diff --git a/python/src/ops.cpp b/python/src/ops.cpp index a1cbca4a8b..b64a4c9eee 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -746,6 +746,9 @@ void init_ops(nb::module_& m) { - All but the last two dimensions of each input are broadcast with one another using standard numpy-style broadcasting semantics. + Only inexact types are supported. The promoted type of ``a`` and ``b`` + must be floating point or complex. + Args: a (array): Input array or scalar. b (array): Input array or scalar. @@ -5927,6 +5930,9 @@ void init_ops(nb::module_& m) { Perform the Einstein summation convention on the operands. + Contractions are implemented with :func:`matmul` and have the same + requirements. + Args: subscripts (str): The Einstein summation convention equation. *operands (array): The input arrays.