support label with float input of cross_entropy, test=develop (Paddle…

…Paddle#30929) * support label with float input of cross_entropy, test=develop * fix code style in nn/functional/loss.py, test=develop
qingqing01 · Feb 9, 2021 · f5ca2db · f5ca2db
1 parent 52edaec
commit f5ca2db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,