Method _check_input_shape

tensorlayer/layers/normalization.py:329–332

(self, inputs)

    """

    def _check_input_shape(self, inputs):
        inputs_shape = _compute_shape(inputs)
        if len(inputs_shape) != 2 and len(inputs_shape) != 3:
            raise ValueError('expected input to be 2D or 3D, but got {}D input'.format(inputs.ndim))


class BatchNorm2d(BatchNorm):