package com.intel.daal.examples.neural_networks;
import com.intel.daal.algorithms.neural_networks.layers.maximum_pooling3d.*;
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;
import com.intel.daal.data_management.data.NumericTable;
class MaxPool3DLayerDenseBatch {
private static DaalContext context = new DaalContext();
static float 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 long[] dims = {3, 2, 4};
public static void main(String[] args) throws java.io.FileNotFoundException, java.io.IOException {
HomogenTensor data = new HomogenTensor(context, dims, dataArray);
long nDim = data.getDimensions().length;
Service.printTensor3d("Forward maximum pooling layer input:", data, 0, 0);
MaximumPooling3dForwardBatch maximumPooling3DLayerForward = new MaximumPooling3dForwardBatch(context, Float.class, MaximumPooling3dMethod.defaultDense, nDim);
maximumPooling3DLayerForward.input.set(ForwardInputId.data, data);
MaximumPooling3dForwardResult forwardResult = maximumPooling3DLayerForward.compute();
Service.printTensor3d("Forward maximum pooling layer result:", forwardResult.get(ForwardResultId.value), 0, 0);
Service.printTensor3d("Forward maximum pooling layer selected indices:",
forwardResult.get(MaximumPooling3dLayerDataId.auxSelectedIndices), 0, 0);
MaximumPooling3dBackwardBatch maximumPooling3DLayerBackward = new MaximumPooling3dBackwardBatch(context, Float.class, MaximumPooling3dMethod.defaultDense, nDim);
maximumPooling3DLayerBackward.input.set(BackwardInputId.inputGradient, forwardResult.get(ForwardResultId.value));
maximumPooling3DLayerBackward.input.set(BackwardInputLayerDataId.inputFromForward,
forwardResult.get(ForwardResultLayerDataId.resultForBackward));
MaximumPooling3dBackwardResult backwardResult = maximumPooling3DLayerBackward.compute();
Service.printTensor3d("Backward maximum pooling layer result:", backwardResult.get(BackwardResultId.gradient), 0, 0);
context.dispose();
}
}