680f4831e6c8f3102be8d99056b746b48ce34051,einops.py,TransformRecipe,reconstruct_from_shape,#TransformRecipe#,62

Before Change

            if axes_lengths[axis] is not None:
                // checking dimension
                if isinstance(axes_lengths[axis], int) and isinstance(length, int):
                    print("checked dimension")
                    // TODO check for static graphs
                    assert axes_lengths[axis] == length
            else:
                axes_lengths[axis] = length

After Change

            for axis in known_axes:
                known_product *= axes_lengths[axis]

            if len(unknown_axes) == 0:
                if isinstance(length, int) and isinstance(known_product, int):
                    assert length == known_product
            else:
                if isinstance(length, int) and isinstance(known_product, int):
                    assert length % known_product == 0
                unknown_axis, = unknown_axes
                axes_lengths[unknown_axis] = length // known_product

        init_shapes = axes_lengths
        reduced_axes_lengths = [dim for i, dim in enumerate(axes_lengths) if i not in self.reduced_elementary_axes]
        final_shapes = []
        for grouping in self.output_composite_axes:

Instances