[TF-numpy] Adds __matmul__ method to ndarray.
PiperOrigin-RevId: 317763209 Change-Id: I3a2ab8e07b040144239ecee58d4ba9267f5b8977
This commit is contained in:
parent
5d4a29eaf5
commit
6b0b4876f8
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue