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

fix bug for TransformedDistribution

[Cherry-pick] fix bug for TransformedDistribution (#46157)
fix bug for TransformedDistribution
a5d4f571 · MayYouBeProsperous · GitHub · 4261ae34 · a5d4f571
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/distribution/transformed_distribution.py python/paddle/distribution/transformed_distribution.py +1 -1

未找到文件。
--- a/python/paddle/distribution/transformed_distribution.py
+++ b/python/paddle/distribution/transformed_distribution.py
@@ -77,7 +77,7 @@ class TransformedDistribution(distribution.Distribution):
            max(len(base.event_shape)-chain._domain.event_rank, 0)
        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:])

    def sample(self, shape=()):
        """Sample from ``TransformedDistribution``.