680f4831e6c8f3102be8d99056b746b48ce34051,einops.py,,reduce,#,243

Before Change

    for axis, axis_length in axes_lengths.items():
        elementary_axes = get_axes_names(axis)
        if len(elementary_axes) == 1 and isinstance(axis_length, int):
            update_axis_length(elementary_axes[0], axis_length)
        else:
            assert len(elementary_axes) == len(axis_length), [elementary_axes, axis_length]
            for c, v in zip(elementary_axes, axis_length):
                if c != "_":
                    update_axis_length(c, v)

    denominator_indices = []
    position_lookup = dict(zip(list(known_lengths), range(len(known_lengths))))
    // inferring rest of sizes from arguments
    for composite_axis in composite_axes_left:

After Change

    for axis, axis_length in axes_lengths.items():
        elementary_axes = get_axes_names(axis)
        // TODO better name validation
        assert len(elementary_axes) == 1
        update_axis_length(elementary_axes[0], axis_length)

    input_axes_known_unknown = []
    // inferring rest of sizes from arguments

Instances