#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;
string datasetFileName = "../data/batch/layer.csv";
const size_t dimension = 1;
int main(int argc, char *argv[])
{
checkArguments(argc, argv, 1, &datasetFileName);
TensorPtr data = readTensorFromCSV(datasetFileName);
printTensor(data, "Forward batch normalization layer input (first 5 rows):", 5);
const Collection<size_t> &dataDims = data->getDimensions();
size_t dimensionSize = dataDims[dimension];
Collection<size_t> dimensionSizes;
dimensionSizes.push_back(dimensionSize);
TensorPtr weights(new HomogenTensor<>(dimensionSizes, Tensor::doAllocate, 1.0f));
TensorPtr biases (new HomogenTensor<>(dimensionSizes, Tensor::doAllocate, 2.0f));
TensorPtr populationMean (new HomogenTensor<>(dimensionSizes, Tensor::doAllocate, 0.0f));
TensorPtr populationVariance(new HomogenTensor<>(dimensionSizes, Tensor::doAllocate, 0.0f));
batch_normalization::forward::Batch<> forwardLayer;
forwardLayer.parameter.dimension = dimension;
forwardLayer.input.set(forward::data, data);
forwardLayer.input.set(forward::weights, weights);
forwardLayer.input.set(forward::biases, biases);
forwardLayer.input.set(batch_normalization::forward::populationMean, populationMean);
forwardLayer.input.set(batch_normalization::forward::populationVariance, populationVariance);
forwardLayer.compute();
batch_normalization::forward::ResultPtr forwardResult = forwardLayer.getResult();
printTensor(forwardResult->get(forward::value), "Forward batch normalization layer result (first 5 rows):", 5);
printTensor(forwardResult->get(batch_normalization::auxMean), "Mini-batch mean (first 5 values):", 5);
printTensor(forwardResult->get(batch_normalization::auxStandardDeviation), "Mini-batch standard deviation (first 5 values):", 5);
printTensor(forwardResult->get(batch_normalization::auxPopulationMean), "Population mean (first 5 values):", 5);
printTensor(forwardResult->get(batch_normalization::auxPopulationVariance), "Population variance (first 5 values):", 5);
TensorPtr inputGradientTensor = TensorPtr(new HomogenTensor<>(dataDims, Tensor::doAllocate, 10.0f));
batch_normalization::backward::Batch<> backwardLayer;
backwardLayer.parameter.dimension = dimension;
backwardLayer.input.set(backward::inputGradient, inputGradientTensor);
backwardLayer.input.set(backward::inputFromForward, forwardResult->get(forward::resultForBackward));
backwardLayer.compute();
backward::ResultPtr backwardResult = backwardLayer.getResult();
printTensor(backwardResult->get(backward::gradient), "Backward batch normalization layer result (first 5 rows):", 5);
printTensor(backwardResult->get(backward::weightDerivatives), "Weight derivatives (first 5 values):", 5);
printTensor(backwardResult->get(backward::biasDerivatives), "Bias derivatives (first 5 values):", 5);
return 0;
}