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