Merge pull request #98 from asmeurer/vecdot-conj

Conjugate the first argument in vecdot
data-apis · Dec 4, 2024 · 179e3e2 · 179e3e2
2 parents 94febbc + e2ae30b
commit 179e3e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/array_api_strict/_linear_algebra_functions.py b/array_api_strict/_linear_algebra_functions.py
@@ -86,5 +86,5 @@ def vecdot(x1: Array, x2: Array, /, *, axis: int = -1) -> Array:
     x1_ = np.moveaxis(x1_, axis, -1)
     x2_ = np.moveaxis(x2_, axis, -1)
 
-    res = x1_[..., None, :] @ x2_[..., None]
+    res = np.conj(x1_[..., None, :]) @ x2_[..., None]
     return Array._new(res[..., 0, 0], device=x1.device)