experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[TF-numpy] Adds __rmatmul__ method to ndarray.

Browse Source 
PiperOrigin-RevId: 317771125
Change-Id: I719c46d97ae1c68ac59dcd1cf8f65d067ddc7658
This commit is contained in:
Peng Wang 2020-06-22 17:47:47 -07:00 committed by TensorFlower Gardener
parent 34b4fab30a
commit e74a115bcf
1 changed files with 2 additions and 1 deletions

3
tensorflow/python/ops/numpy_ops/np_math_ops.py
View File

@ -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):