c571066f4b0cb46f5aef7f4fa280e1ffabb24172,theano/tensor/opt_uncanonicalize.py,,local_alloc_dimshuffle,#Any#,106

Before Change

            new_order = input_.owner.op.new_order
            flag = False
            for i, dim in enumerate(new_order_bool):
                if i == 0 and dim == "x":
                    flag = True
                elif dim == "x" and flag:
                    continue
                elif i > 0 and flag:
                    flag = False
                elif i > 0 and not dim == "x":
                    continue
                else:
                    return False
            return input_.owner.inputs
    return False

After Change

            // check if it only adds dimension to the left
            new_order = input_.owner.op.new_order
            expected_new_order = ("x",) * (input_.ndim - input_.owner.inputs[0].ndim) + \
                    tuple(range(input_.owner.inputs.ndim))
            if new_order != expected_new_order:
                return False
            return input_.owner.inputs

Instances