fix bug for TransformedDistribution (#46035)

71c5ec87 · MayYouBeProsperous · GitHub · 2a5bd7dc · 71c5ec87
显示空白变更内容
内联并排

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``.