#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;
int main()
{
Collection<size_t> inDims;
inDims.push_back(2);
inDims.push_back(1);
inDims.push_back(3);
inDims.push_back(4);
TensorPtr tensorData = TensorPtr(new HomogenTensor<>(inDims, Tensor::doAllocate, 1.0f));
lcn::forward::Batch<> lcnLayerForward;
lcnLayerForward.input.set(forward::data, tensorData);
lcnLayerForward.compute();
lcn::forward::ResultPtr forwardResult = lcnLayerForward.getResult();
printTensor(forwardResult->get(forward::value), "Forward local contrast normalization layer result:");
printTensor(forwardResult->get(lcn::auxCenteredData), "Centered data tensor:");
printTensor(forwardResult->get(lcn::auxSigma), "Sigma tensor:");
printTensor(forwardResult->get(lcn::auxC), "C tensor:");
printTensor(forwardResult->get(lcn::auxInvMax), "Inverted max(sigma, C):");
TensorPtr tensorDataBack = TensorPtr(new HomogenTensor<>(inDims, Tensor::doAllocate, 0.01f));
lcn::backward::Batch<> lcnLayerBackward;
lcnLayerBackward.input.set(backward::inputGradient, tensorDataBack);
lcnLayerBackward.input.set(backward::inputFromForward, forwardResult->get(forward::resultForBackward));
lcnLayerBackward.compute();
backward::ResultPtr backwardResult = lcnLayerBackward.getResult();
printTensor(backwardResult->get(backward::gradient), "Local contrast normalization layer backpropagation gradient result:");
return 0;
}