def compute_output_shape()

in code/source/custom_layer.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

    def compute_output_shape(self, input_shape):
        if self.output_representation == 'pooled_output':
            return (None, 768)
        else:
            return (None, None, 768)