support label with float input of cross_entropy, test=develop (#30929)

* support label with float input of cross_entropy, test=develop * fix code style in nn/functional/loss.py, test=develop
4 years ago · f5ca2db2cc
parent 52edaecc5d
commit f5ca2db2cc
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/nn/functional/loss.py
+++ b/python/paddle/nn/functional/loss.py
@ -1275,7 +1275,8 @@ def cross_entropy(input,
    fluid.data_feeder.check_variable_and_dtype(
        input, 'input', ['float32', 'float64'], 'softmax_cross_entropy')
    fluid.data_feeder.check_variable_and_dtype(
-        label, 'label', ['int32', 'int64'], 'softmax_cross_entropy')
+        label, 'label', ['int32', 'int64', 'float32', 'float64'],
+        'softmax_cross_entropy')
    out = softmax_with_cross_entropy(
        input,
        label,