From e74a115bcf5cd27f476b46161a639e9ec599491d Mon Sep 17 00:00:00 2001 From: Peng Wang Date: Mon, 22 Jun 2020 17:47:47 -0700 Subject: [PATCH] [TF-numpy] Adds __rmatmul__ method to ndarray. PiperOrigin-RevId: 317771125 Change-Id: I719c46d97ae1c68ac59dcd1cf8f65d067ddc7658 --- tensorflow/python/ops/numpy_ops/np_math_ops.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/ops/numpy_ops/np_math_ops.py b/tensorflow/python/ops/numpy_ops/np_math_ops.py index 427aa96e5a4..03329bbdbf1 100644 --- a/tensorflow/python/ops/numpy_ops/np_math_ops.py +++ b/tensorflow/python/ops/numpy_ops/np_math_ops.py @@ -950,11 +950,12 @@ setattr(np_arrays.ndarray, '__sub__', _wrap(subtract)) setattr(np_arrays.ndarray, '__rsub__', _wrap(subtract, True)) setattr(np_arrays.ndarray, '__mul__', _wrap(multiply)) setattr(np_arrays.ndarray, '__rmul__', _wrap(multiply, True)) +setattr(np_arrays.ndarray, '__matmul__', _wrap(matmul)) +setattr(np_arrays.ndarray, '__rmatmul__', _wrap(matmul, True)) setattr(np_arrays.ndarray, '__pow__', _wrap(power)) setattr(np_arrays.ndarray, '__rpow__', _wrap(power, True)) setattr(np_arrays.ndarray, '__truediv__', _wrap(true_divide)) setattr(np_arrays.ndarray, '__rtruediv__', _wrap(true_divide, True)) -setattr(np_arrays.ndarray, '__matmul__', _wrap(matmul)) def _comparison(tf_fun, x1, x2, cast_bool_to_int=False):