Method shape

mla/neuralnet/layers/convnet.py:122–126

(self, x_shape)

        )

    def shape(self, x_shape):
        h, w = convoltuion_shape(
            x_shape[2], x_shape[3], self.pool_shape, self.stride, self.padding
        )
        return x_shape[0], x_shape[1], h, w


class Flatten(Layer):