[TF-numpy] Adds __matmul__ method to ndarray.

PiperOrigin-RevId: 317763209
Change-Id: I3a2ab8e07b040144239ecee58d4ba9267f5b8977
This commit is contained in:
Peng Wang 2020-06-22 16:54:12 -07:00 committed by TensorFlower Gardener
parent 5d4a29eaf5
commit 6b0b4876f8
1 changed files with 1 additions and 0 deletions

View File

@ -954,6 +954,7 @@ 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):