p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Check number validation of power in powertransform

This commit is contained in:
peixu_ren 2020-08-19 15:26:43 -04:00
parent b11ef57b65
commit 5acf10941d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/probability/bijector/power_transform.py
View File

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