From ae50dd86fdb80840d7798cc39d431a1c82da56e5 Mon Sep 17 00:00:00 2001 From: danishnxt Date: Wed, 11 Nov 2020 16:43:31 -0500 Subject: [PATCH] TensorDot docString fix --- mindspore/ops/composite/math_ops.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/ops/composite/math_ops.py b/mindspore/ops/composite/math_ops.py index 169a008604..6a97ecbc22 100644 --- a/mindspore/ops/composite/math_ops.py +++ b/mindspore/ops/composite/math_ops.py @@ -185,6 +185,10 @@ def TensorDot(x1, x2, axes): >>> input_x1 = Tensor(np.ones(shape=[1, 2, 3]), mindspore.float32) >>> input_x2 = Tensor(np.ones(shape=[3, 1, 2]), mindspore.float32) >>> output = C.TensorDot(input_x1, input_x2, ((0,1),(1,2))) + >>> print(output) + [[2,2,2], + [2,2,2], + [2,2,2]] """ shape_op = P.Shape() reshape_op = P.Reshape()