BUG: fix up matmul

Author: ev-br

torch_np/_binary_ufuncs.py

@@ -68,7 +68,7 @@ def matmul(
     extobj=None,
     axes=None,
     axis=None,
-) -> OutArray:
+):
     tensors = _helpers.ufunc_preprocess(
         (x1, x2), out, True, casting, order, dtype, subok, signature, extobj
     )
@@ -77,7 +77,7 @@ def matmul(
 
     # NB: do not broadcast input tensors against the out=... array
     result = _binary_ufuncs.matmul(*tensors)
-    return result, out
+    return _helpers.result_or_out(result, out)
 
 
 #

torch_np/tests/numpy_tests/core/test_multiarray.py

@@ -5979,7 +5979,7 @@ def test_out_contiguous(self):
         # test out non-contiguous
         out = np.ones((5, 2, 2), dtype=float)
         c = self.matmul(a, b, out=out[..., 0])
-        assert c._tensor._base is out._tensor     # FIXME: self.tensor (no underscore)
+        assert c.tensor._base is out.tensor
         assert_array_equal(c, tgt)
         c = self.matmul(a, v, out=out[:, 0, 0])
         assert_array_equal(c, tgt_mv)