Browse Source

TensorDot docString fix

tags/v1.1.0
danishnxt 5 years ago
parent
commit
ae50dd86fd
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ops/composite/math_ops.py

+ 4
- 0
mindspore/ops/composite/math_ops.py View File

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


Loading…
Cancel
Save