!5212 Quick fix scalar add flaws in MindSpore

Merge pull request !5212 from peixu_ren/custom_bijector
pull/5212/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 5d091dca9f

@ -69,6 +69,7 @@ class ScalarAffine(Bijector):
param=param)
self.abs = P.Abs()
self.oneslike = P.OnesLike()
self.log = log_generic
@property
@ -92,7 +93,7 @@ class ScalarAffine(Bijector):
f(x) = a * x + b
"""
x = self._check_value(x, 'value')
return self.scale * x + self.shift
return self.scale * x + self.shift * self.oneslike(x)
def _inverse(self, y):
r"""

Loading…
Cancel
Save