package com.intel.daal.examples.neural_networks;
import com.intel.daal.algorithms.neural_networks.layers.locallyconnected2d.*;
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.ForwardInputId;
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 Locallycon2DLayerDenseBatch {
private static DaalContext context = new DaalContext();
public static void main(String[] args) throws java.io.FileNotFoundException, java.io.IOException {
long[] dimensionSizes = new long[4];
dimensionSizes[0] = 2;
dimensionSizes[1] = 2;
dimensionSizes[2] = 6;
dimensionSizes[3] = 8;
double[] data = new double[192];
Tensor dataTensor = new HomogenTensor(context, dimensionSizes, data, 1.0);
LocallyConnected2dForwardBatch locallyconnected2DLayerForward = new LocallyConnected2dForwardBatch(context, Float.class, LocallyConnected2dMethod.defaultDense);
locallyconnected2DLayerForward.input.set(ForwardInputId.data, dataTensor);
LocallyConnected2dForwardResult forwardResult = locallyconnected2DLayerForward.compute();
Service.printTensor("Forward two-dimensional locally connected layer result (first 5 rows):", forwardResult.get(ForwardResultId.value), 5, 0);
Service.printTensor("Two-dimensional locally connected layer weights (first 5 rows):", forwardResult.get(LocallyConnected2dLayerDataId.auxWeights), 5, 0);
int nSize = (int)forwardResult.get(ForwardResultId.value).getSize();
long[] dims = forwardResult.get(ForwardResultId.value).getDimensions();
double[] backData = new double[nSize];
Tensor tensorDataBack = new HomogenTensor(context, dims, backData, 0.01);
LocallyConnected2dBackwardBatch locallyconnected2DLayerBackward = new LocallyConnected2dBackwardBatch(context, Float.class, LocallyConnected2dMethod.defaultDense);
locallyconnected2DLayerBackward.input.set(BackwardInputId.inputGradient, tensorDataBack);
locallyconnected2DLayerBackward.input.set(BackwardInputLayerDataId.inputFromForward,
forwardResult.get(ForwardResultLayerDataId.resultForBackward));
LocallyConnected2dBackwardResult backwardResult = locallyconnected2DLayerBackward.compute();
Service.printTensor("Backward 2D locally connected layer result (first 5 rows):", backwardResult.get(BackwardResultId.gradient), 5, 0);
Service.printTensor("Weights derivative (first 5 rows):", backwardResult.get(BackwardResultId.weightDerivatives), 5, 0);
Service.printTensor("Biases derivative (first 5 rows):", backwardResult.get(BackwardResultId.biasDerivatives), 5, 0);
context.dispose();
}
}