Fix sample code of poisson_nll_loss (#53551)

3fd2e765 · LyndonKong · GitHub · acefdeb7 · 3fd2e765
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/nn/functional/loss.py python/paddle/nn/functional/loss.py +1 -1

未找到文件。
--- a/python/paddle/nn/functional/loss.py
+++ b/python/paddle/nn/functional/loss.py
@@ -1541,7 +1541,7 @@ def poisson_nll_loss(

            input = paddle.randn([5, 2], dtype=paddle.float32)
            label = paddle.randn([5, 2], dtype=paddle.float32)
-            loss = F.poisson_nll_loss(input, label, log_input=True, reduction='None')
+            loss = F.poisson_nll_loss(input, label, log_input=True, reduction='none')
            print(loss)
            loss = F.poisson_nll_loss(input, label, reduction='mean')
            print(loss)