Update: Cross Entropy Torch Frontend (ivy-llc#25881)

iababio · Sep 27, 2023 · d2e81d7 · d2e81d7
1 parent a034740
commit d2e81d7
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/ivy/functional/frontends/torch/nn/functional/loss_functions.py b/ivy/functional/frontends/torch/nn/functional/loss_functions.py
@@ -212,12 +212,7 @@ def cross_entropy(
     reduction="mean",
     label_smoothing=0.0,
 ):
-    input = ivy.softmax(input)
-    ret = ivy.cross_entropy(target, input, epsilon=label_smoothing)
-    if weight is not None:
-        ret = ivy.multiply(weight, ret)
-    ret = _apply_reduction(reduction, size_average, reduce, ret)
-    return ret
+    return ivy.cross_entropy(target, input, epsilon=label_smoothing)
 
 
 @to_ivy_arrays_and_back