diff --git a/mindspore/nn/probability/bijector/power_transform.py b/mindspore/nn/probability/bijector/power_transform.py index 456f635818..1d9d2c5a88 100644 --- a/mindspore/nn/probability/bijector/power_transform.py +++ b/mindspore/nn/probability/bijector/power_transform.py @@ -15,6 +15,7 @@ """Power Bijector""" from mindspore.ops import operations as P from mindspore._checkparam import Validator as validator +from mindspore._checkparam import Rel from .bijector import Bijector class PowerTransform(Bijector): @@ -53,6 +54,7 @@ class PowerTransform(Bijector): param = dict(locals()) if param is None else param super(PowerTransform, self).__init__(name=name, param=param) validator.check_value_type('power', power, [int, float], self.name) + validator.check_number("power", power, 0, Rel.GE, self.name) self._power = power self.pow = P.Pow() self.exp = P.Exp()