diff --git a/python/paddle/nn/functional/loss.py b/python/paddle/nn/functional/loss.py index b105d3d7fc28e8..ae7b8813a0e3e3 100644 --- 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)