未验证 提交 a5d4f571 编写于 作者: M MayYouBeProsperous 提交者: GitHub

[Cherry-pick] fix bug for TransformedDistribution (#46157)

fix bug for TransformedDistribution
上级 4261ae34
...@@ -77,7 +77,7 @@ class TransformedDistribution(distribution.Distribution): ...@@ -77,7 +77,7 @@ class TransformedDistribution(distribution.Distribution):
max(len(base.event_shape)-chain._domain.event_rank, 0) max(len(base.event_shape)-chain._domain.event_rank, 0)
super(TransformedDistribution, self).__init__( super(TransformedDistribution, self).__init__(
transformed_shape[:len(transformed_shape) - transformed_event_rank], transformed_shape[:len(transformed_shape) - transformed_event_rank],
transformed_shape[:len(transformed_shape) - transformed_event_rank]) transformed_shape[len(transformed_shape) - transformed_event_rank:])
def sample(self, shape=()): def sample(self, shape=()):
"""Sample from ``TransformedDistribution``. """Sample from ``TransformedDistribution``.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册