58 from daal.algorithms.neural_networks
import layers
59 from daal.algorithms.neural_networks.layers
import spatial_stochastic_pooling2d
60 from daal.data_management
import HomogenTensor
62 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
63 if utils_folder
not in sys.path:
64 sys.path.insert(0, utils_folder)
65 from utils
import printTensor
69 dataArray = np.array([[[[1, 2, 3, 4],
78 [130, 140, 150, 160]],
79 [[170, 180, 190, 200],
80 [210, 220, 230, 240]]]],
83 if __name__ ==
"__main__":
84 data = HomogenTensor(dataArray)
86 printTensor(data,
"Forward two-dimensional spatial pyramid stochastic pooling layer input (first 10 rows):", 10)
89 forwardLayer = spatial_stochastic_pooling2d.forward.Batch(2, nDim)
90 forwardLayer.input.setInput(layers.forward.data, data)
93 forwardLayer.compute()
96 forwardResult = forwardLayer.getResult()
98 printTensor(forwardResult.getResult(layers.forward.value),
"Forward two-dimensional spatial pyramid stochastic pooling layer result (first 5 rows):", 5)
99 printTensor(forwardResult.getLayerData(layers.spatial_stochastic_pooling2d.auxSelectedIndices),
100 "Forward two-dimensional spatial pyramid stochastic pooling layer selected indices (first 10 rows):", 10)
103 backwardLayer = layers.spatial_stochastic_pooling2d.backward.Batch(2, nDim)
104 backwardLayer.input.setInput(layers.backward.inputGradient, forwardResult.getResult(layers.forward.value))
105 backwardLayer.input.setInputLayerData(layers.backward.inputFromForward, forwardResult.getResultLayerData(layers.forward.resultForBackward))
108 backwardLayer.compute()
111 backwardResult = backwardLayer.getResult()
113 printTensor(backwardResult.getResult(layers.backward.gradient),
114 "Backward two-dimensional spatial pyramid stochastic pooling layer result (first 10 rows):", 10)