diff --git a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/loss/BaseLoss.java b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/loss/BaseLoss.java
index <HASH>..<HASH> 100644
--- a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/loss/BaseLoss.java
+++ b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/loss/BaseLoss.java
@@ -71,10 +71,7 @@ public abstract class BaseLoss extends DynamicCustomOp {
 
     public abstract String opName();
 
-    @Override
-    public int getNumOutputs() {
-        return 1;
-    }
+ 
 
     @Override
     public List<DataType> calculateOutputDataTypes(List<DataType> inputDataTypes){
Update BaseLoss.java