Browse Source

Fix ReduceMax op bool type problem.

tags/v1.1.0
liangchenghui 5 years ago
parent
commit
618c2fd523
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ops/operations/math_ops.py

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

@@ -497,6 +497,8 @@ class ReduceMax(_Reduce):
super(ReduceMax, self).__init__(keep_dims)
self.__setattr_flag__ = True

def __infer__(self, input_x, axis):
return self.do_infer(input_x, axis, mstype.number_type + (mstype.bool_,))

class ReduceMin(_Reduce):
"""


Loading…
Cancel
Save