Browse Source

Getting Some simple issues fixed

tags/v0.2.0-alpha
zhangz0911gm 5 years ago
parent
commit
92ab989a85
2 changed files with 4 additions and 4 deletions
  1. +2
    -2
      mindspore/ops/operations/math_ops.py
  2. +2
    -2
      mindspore/ops/operations/other_ops.py

+ 2
- 2
mindspore/ops/operations/math_ops.py View File

@@ -773,8 +773,8 @@ class Mul(_MathBinaryOp):
Tensor, the shape is same as the shape after broadcasting, and the data type is same as 'input_x'.

Examples:
>>> input_x = Tensor(np.array([1, 2, 3]), mindspore.int32)
>>> input_y = Tensor(np.array([4, 5, 6]), mindspore.int32)
>>> input_x = Tensor(np.array([1.0, 2.0, 3.0]), mindspore.float32)
>>> input_y = Tensor(np.array([4.0, 5.0, 6.0]), mindspore.float32)
>>> mul = P.Mul()
>>> mul(input_x, input_y)
[4, 10, 18]


+ 2
- 2
mindspore/ops/operations/other_ops.py View File

@@ -209,8 +209,8 @@ class IOU(PrimitiveWithInfer):

Examples:
>>> iou = P.IOU()
>>> anchor_boxes = Tensor(np.random.randint(1,5, [10, 4]))
>>> gt_boxes = Tensor(np.random.randint(1,5, [3, 4]))
>>> anchor_boxes = Tensor(np.random.randint(1.0, 5.0, [3, 4]), mindspore.float32)
>>> gt_boxes = Tensor(np.random.randint(1.0, 5.0, [3, 4]), mindspore.float32)
>>> iou(anchor_boxes, gt_boxes)
"""



Loading…
Cancel
Save