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

[TF-numpy] Adds __matmul__ method to ndarray.

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/ops/numpy_ops/np_math_ops.py
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, '__rpow__', _wrap(power, True))
setattr(np_arrays.ndarray, '__truediv__', _wrap(true_divide)) setattr(np_arrays.ndarray, '__truediv__', _wrap(true_divide))
setattr(np_arrays.ndarray, '__rtruediv__', _wrap(true_divide, True)) 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): def _comparison(tf_fun, x1, x2, cast_bool_to_int=False):