34 from daal.algorithms.neural_networks
import layers
35 from daal.algorithms.neural_networks.layers
import spatial_maximum_pooling2d
36 from daal.data_management
import HomogenTensor
38 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
39 if utils_folder
not in sys.path:
40 sys.path.insert(0, utils_folder)
41 from utils
import printTensor
45 dataArray = np.array([[[[1, 2, 3, 4],
54 [-13, -14, -15, -16]],
55 [[-17, -18, -19, -20],
56 [-21, -22, -23, -24]]]],
59 if __name__ ==
"__main__":
60 data = HomogenTensor(dataArray)
63 printTensor(data,
"Forward two-dimensional spatial pyramid maximum pooling layer input (first 10 rows):", 10)
66 forwardLayer = spatial_maximum_pooling2d.forward.Batch(2, nDim)
67 forwardLayer.input.setInput(layers.forward.data, data)
70 forwardLayer.compute()
73 forwardResult = forwardLayer.getResult()
75 printTensor(forwardResult.getResult(layers.forward.value),
"Forward two-dimensional spatial pyramid maximum pooling layer result (first 5 rows):", 5)
76 printTensor(forwardResult.getLayerData(layers.spatial_maximum_pooling2d.auxSelectedIndices),
77 "Forward two-dimensional spatial pyramid maximum pooling layer selected indices (first 10 rows):", 10)
80 backwardLayer = layers.spatial_maximum_pooling2d.backward.Batch(2, nDim)
81 backwardLayer.input.setInput(layers.backward.inputGradient, forwardResult.getResult(layers.forward.value))
82 backwardLayer.input.setInputLayerData(layers.backward.inputFromForward, forwardResult.getResultLayerData(layers.forward.resultForBackward))
85 backwardLayer.compute()
88 backwardResult = backwardLayer.getResult()
90 printTensor(backwardResult.getResult(layers.backward.gradient),
91 "Backward two-dimensional spatial pyramid maximum pooling layer result (first 10 rows):", 10)