package com.intel.daal.examples.neural_networks;
import com.intel.daal.algorithms.neural_networks.layers.softmax_cross.*;
import com.intel.daal.algorithms.neural_networks.layers.ForwardResultId;
import com.intel.daal.algorithms.neural_networks.layers.ForwardResultLayerDataId;
import com.intel.daal.algorithms.neural_networks.layers.loss.LossForwardInputId;
import com.intel.daal.algorithms.neural_networks.layers.BackwardResultId;
import com.intel.daal.algorithms.neural_networks.layers.BackwardInputId;
import com.intel.daal.algorithms.neural_networks.layers.BackwardInputLayerDataId;
import com.intel.daal.data_management.data.Tensor;
import com.intel.daal.data_management.data.HomogenTensor;
import com.intel.daal.examples.utils.Service;
import com.intel.daal.services.DaalContext;
class LossSoftmaxEntrLayerDenseBatch {
private static DaalContext context = new DaalContext();
static double dataArray[] = { 1, 2, 3, 4,
5, 6, 7, 8,
9, 10, 11, 12,
13, 14, 15, 16,
17, 18, 19, 20,
21, 22, 23, 24};
static float groundTruthArray[] = { 1, 0, 0, 1, 0, 0, 1, 1, 1, 0, 0, 1};
static long[] dims = {3, 2, 4};
static long[] gtDims = {3, 1, 4};
public static void main(String[] args) throws java.io.FileNotFoundException, java.io.IOException {
HomogenTensor data = new HomogenTensor(context, dims, dataArray);
HomogenTensor groundTruth = new HomogenTensor(context, gtDims, groundTruthArray);
Service.printTensor("Forward softmax cross-entropy layer input data:", data, 0, 0);
Service.printTensor("Forward softmax cross-entropy layer input ground truths:", groundTruth, 0, 0);
SoftmaxCrossForwardBatch forwardLayer = new SoftmaxCrossForwardBatch(context, Float.class, SoftmaxCrossMethod.defaultDense);
forwardLayer.input.set(LossForwardInputId.data, data);
forwardLayer.input.set(LossForwardInputId.groundTruth, groundTruth);
SoftmaxCrossForwardResult forwardResult = forwardLayer.compute();
Service.printTensor("Forward softmax cross-entropy layer result (first 5 rows):", forwardResult.get(ForwardResultId.value), 5, 0);
Service.printTensor("Softmax Cross-Entropy layer probabilities estimations (first 5 rows):", forwardResult.get(SoftmaxCrossLayerDataId.auxProbabilities), 5, 0);
Service.printTensor("Softmax Cross-Entropy layer ground truth (first 5 rows):", forwardResult.get(SoftmaxCrossLayerDataId.auxGroundTruth), 5, 0);
SoftmaxCrossBackwardBatch backwardLayer = new SoftmaxCrossBackwardBatch(context, Float.class, SoftmaxCrossMethod.defaultDense);
backwardLayer.input.set(BackwardInputLayerDataId.inputFromForward, forwardResult.get(ForwardResultLayerDataId.resultForBackward));
SoftmaxCrossBackwardResult backwardResult = backwardLayer.compute();
Service.printTensor("Backward softmax cross-entropy layer result (first 5 rows):", backwardResult.get(BackwardResultId.gradient), 5, 0);
context.dispose();
}
}