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

[TF-numpy] Fixed einsum's signature to be Python2 compatible.

Browse Source 
PiperOrigin-RevId: 317490711
Change-Id: Ia33d3958bb3163131d40262cd7247411be4ade52
This commit is contained in:
Peng Wang 2020-06-20 15:46:53 -07:00 committed by TensorFlower Gardener
parent ae423bd3bf
commit 793c5ca70b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1340,7 +1340,9 @@ def meshgrid(*xi, **kwargs):
@np_utils.np_doc('einsum')
def einsum(subscripts, *operands, casting='safe', optimize=False): # pylint: disable=missing-docstring
def einsum(subscripts, *operands, **kwargs): # pylint: disable=missing-docstring
casting = kwargs.get('casting', 'safe')
optimize = kwargs.get('optimize', False)
if casting == 'safe':
operands = np_array_ops._promote_dtype(*operands) # pylint: disable=protected-access
elif casting == 'no':