From 793c5ca70b26763c4f85f3c2961d82653d665caf Mon Sep 17 00:00:00 2001 From: Peng Wang Date: Sat, 20 Jun 2020 15:46:53 -0700 Subject: [PATCH] [TF-numpy] Fixed einsum's signature to be Python2 compatible. PiperOrigin-RevId: 317490711 Change-Id: Ia33d3958bb3163131d40262cd7247411be4ade52 --- tensorflow/python/ops/numpy_ops/np_math_ops.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/ops/numpy_ops/np_math_ops.py b/tensorflow/python/ops/numpy_ops/np_math_ops.py index f9d3b34e90d..b242e51a2e5 100644 --- a/tensorflow/python/ops/numpy_ops/np_math_ops.py +++ b/tensorflow/python/ops/numpy_ops/np_math_ops.py @@ -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':