c2115ea5f462acdff07919f68ecc77339b47e486,autokeras/blocks/reduction.py,Merge,_inputs_same_shape,#Merge#,84

Before Change

    def _inputs_same_shape(self, inputs):
        for input_node in inputs:
            if input_node.shape.as_list() != inputs[0].shape.as_list():
                return False
        return True


class Flatten(block_module.Block):

After Change

        return layers.Concatenate()(inputs)

    def _inputs_same_shape(self, inputs):
        return all(
            input_node.shape.as_list() == inputs[0].shape.as_list()
            for input_node in inputs
        )


class Flatten(block_module.Block):
    Flatten the input tensor with Keras Flatten layer.

Instances