Merge pull request #201 from asmeurer/vecdot-conj

Conjugate the first argument in vecdot
data-apis · Nov 20, 2024 · ee25aae · ee25aae
2 parents 382bbac + 89a85a6
commit ee25aae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/array_api_compat/common/_aliases.py b/array_api_compat/common/_aliases.py
@@ -479,7 +479,7 @@ def vecdot(x1: ndarray, x2: ndarray, /, xp, *, axis: int = -1) -> ndarray:
     x2_ = xp.moveaxis(x2, axis, -1)
     x1_, x2_ = _broadcast(x1_, x2_)
 
-    res = x1_[..., None, :] @ x2_[..., None]
+    res = xp.conj(x1_[..., None, :]) @ x2_[..., None]
     return res[..., 0, 0]
 
 # isdtype is a new function in the 2022.12 array API specification.