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

!4894 Fix minor bugs in transformed distribution 

Merge pull request !4894 from peixu_ren/custom_bijector
This commit is contained in:
mindspore-ci-bot 2020-08-21 14:11:30 +08:00 committed by Gitee
parent 6782325bfd 98e2a48e4c
commit ec1cf059a7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/nn/probability/distribution/transformed_distribution.py
View File

@ -105,7 +105,7 @@ class TransformedDistribution(Distribution):
return self.exp(self._log_prob(*args, **kwargs))
def _sample(self, *args, **kwargs):
org_sample = self.distribution("sample", shape)
org_sample = self.distribution("sample", *args, **kwargs)
return self.bijector("forward", org_sample)
def _mean(self, *args, **kwargs):
@ -114,6 +114,6 @@ class TransformedDistribution(Distribution):
This function maybe overridden by derived class.
"""
if not self.is_linear_transformation:
raise_not_impl_error(mean)
raise_not_impl_error("mean")
return self.bijector("forward", self.distribution("mean"))