com.intel.analytics.zoo.pipeline.api.autograd
function to calculate the loss (yTrue, yPred) => loss
the pred shape without batch
the target shape without batch which is the same as yPredShape by default.
average the batch result or not