#include "daal.h"
#include "service.h"
using namespace std;
using namespace daal;
using namespace daal::algorithms;
using namespace daal::algorithms::neural_networks::layers;
using namespace daal::data_management;
using namespace daal::services;
static const size_t nDim = 3;
static const size_t dims[] = {3, 2, 4};
static const size_t gtDims[] = {3, 1, 4};
static float gTArray[3][1][4] = {{{1, 0, 0, 1}},
{{0, 0, 1, 1}},
{{1, 0, 0, 1}}};
static float dataArray[3][2][4] = {{{ 1, 2, 3, 4},
{ 5, 6, 7, 8}},
{{9, 10, 11, 12},
{13, 14, 15, 16}},
{{17, 18, 19, 20},
{21, 22, 23, 24}}};
int main()
{
TensorPtr tensorData(new HomogenTensor<>(nDim, dims, (float *)dataArray));
TensorPtr groundTruth(new HomogenTensor<>(nDim, gtDims, (float *)gTArray));
printTensor(tensorData, "Forward softmax cross-entropy layer input data:");
printTensor(groundTruth, "Forward softmax cross-entropy layerr input ground truth:");
loss::softmax_cross::forward::Batch<> softmaxCrossEntropyLayerForward;
softmaxCrossEntropyLayerForward.input.set(forward::data, tensorData);
softmaxCrossEntropyLayerForward.input.set(loss::forward::groundTruth, groundTruth);
softmaxCrossEntropyLayerForward.compute();
loss::softmax_cross::forward::ResultPtr forwardResult = softmaxCrossEntropyLayerForward.getResult();
printTensor(forwardResult->get(forward::value), "Forward softmax cross-entropy layer result (first 5 rows):", 5);
printTensor(forwardResult->get(loss::softmax_cross::auxProbabilities), "Softmax Cross-Entropy layer probabilities estimations (first 5 rows):", 5);
printTensor(forwardResult->get(loss::softmax_cross::auxGroundTruth), "Softmax Cross-Entropy layer ground truth (first 5 rows):", 5);
loss::softmax_cross::backward::Batch<> softmaxCrossEntropyLayerBackward;
softmaxCrossEntropyLayerBackward.input.set(backward::inputFromForward, forwardResult->get(forward::resultForBackward));
softmaxCrossEntropyLayerBackward.compute();
backward::ResultPtr backwardResult = softmaxCrossEntropyLayerBackward.getResult();
printTensor(backwardResult->get(backward::gradient), "Backward softmax cross-entropy layer result (first 5 rows):", 5);
return 0;
}