+
+ // Affine transform layers expect that there is at least
+ // ceil_to_multiple(OutputDimensions, 32) initialized values.
+ // We cannot do this in the affine transform because it requires
+ // preallocating space here.
+ for (IndexType i = OutputDimensions; i < PaddedOutputDimensions; ++i) {
+ output[i] = 0;
+ }
+