{ "package": "einops", "version": "0.8.2", "stubs_only": "no", "py_typed": "YES", "pypi": { "upload_time": "2026-01-26T04:13:17.638853Z", "requires_python": ">=3.9", "size": 56261, "sha256": "609da665570e5e265e27283aab09e7f279ade90c4f01bcfca111f3d3e13f2827" }, "metadata": { "Metadata-Version": [ "2.4" ], "Name": [ "einops" ], "Version": [ "0.8.2" ], "Summary": [ "A new flavour of deep learning operations" ], "Project-URL": [ "Homepage, https://github.com/arogozhnikov/einops" ], "Author": [ "Alex Rogozhnikov" ], "License": [ "MIT" ], "License-File": [ "LICENSE" ], "Keywords": [ "deep learning,einops,machine learning,neural networks,scientific computations,tensor manipulation" ], "Classifier": [ "Intended Audience :: Science/Research", "License :: OSI Approved :: MIT License", "Programming Language :: Python :: 3" ], "Requires-Python": [ ">=3.9" ], "Description-Content-Type": [ "text/markdown" ] }, "module_reports": [ { "path": "einops/einops.py", "symbol_reports": [ { "kind": "function", "name": "einops.einops.reduce", "n_typed": 4, "n_any": 1, "n_untyped": 0, "n_overloads": 2, "n_typable": 5, "n_params": 4, "n_function_overloads": 2, "n_function_params": 4 }, { "kind": "function", "name": "einops.einops.repeat", "n_typed": 3, "n_any": 1, "n_untyped": 0, "n_overloads": 2, "n_typable": 4, "n_params": 3, "n_function_overloads": 2, "n_function_params": 3 }, { "kind": "function", "name": "einops.einops.rearrange", "n_typed": 3, "n_any": 1, "n_untyped": 0, "n_overloads": 2, "n_typable": 4, "n_params": 3, "n_function_overloads": 2, "n_function_params": 3 }, { "kind": "function", "name": "einops.einops.asnumpy", "n_typed": 1, "n_any": 1, "n_untyped": 0, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "einops.einops.einsum", "n_typed": 6, "n_any": 0, "n_untyped": 0, "n_overloads": 4, "n_typable": 6, "n_params": 5, "n_function_overloads": 4, "n_function_params": 5 }, { "kind": "function", "name": "einops.einops.parse_shape", "n_typed": 3, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 3, "n_params": 2, "n_function_overloads": 1, "n_function_params": 2 }, { "kind": "attr", "name": "einops.einops.Reduction", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "einops.einops.CookedRecipe", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "einops.einops.Tensor", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "class", "name": "einops.einops.TransformRecipe", "methods": [ { "kind": "function", "name": "TransformRecipe.__init__", "n_typed": 7, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 8, "n_params": 7, "n_function_overloads": 1, "n_function_params": 7 } ], "properties": [], "attrs": [ { "kind": "attr", "name": "TransformRecipe.elementary_axes_lengths", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "TransformRecipe.axis_name2elementary_axis", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "TransformRecipe.input_composition_known_unknown", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "TransformRecipe.axes_permutation", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "TransformRecipe.first_reduced_axis", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "TransformRecipe.added_axes", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "TransformRecipe.output_composite_axes", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 } ], "n_typable": 15, "n_typed": 14, "n_any": 0, "n_untyped": 1, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 1, "n_method_overloads": 1, "n_method_params": 7, "n_attrs": 7, "n_properties": 0 }, { "kind": "attr", "name": "einops.einops.HashableAxesLengths", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "einops.einops.ReductionCallable", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "einops.einops.FakeHashableAxesLengths", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 } ], "type_ignores": [], "name": "einops.einops", "names": [ "einops.einops.CookedRecipe", "einops.einops.FakeHashableAxesLengths", "einops.einops.HashableAxesLengths", "einops.einops.Reduction", "einops.einops.ReductionCallable", "einops.einops.Tensor", "einops.einops.TransformRecipe", "einops.einops.asnumpy", "einops.einops.einsum", "einops.einops.parse_shape", "einops.einops.rearrange", "einops.einops.reduce", "einops.einops.repeat" ], "n_typable": 44, "n_typed": 39, "n_any": 4, "n_untyped": 1, "n_functions": 6, "n_function_overloads": 12, "n_function_params": 18, "n_methods": 1, "n_method_overloads": 1, "n_method_params": 7, "n_classes": 1, "n_attrs": 13, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/packing.py", "symbol_reports": [ { "kind": "function", "name": "einops.packing.pack", "n_typed": 3, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 3, "n_params": 2, "n_function_overloads": 1, "n_function_params": 2 }, { "kind": "function", "name": "einops.packing.unpack", "n_typed": 4, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 4, "n_params": 3, "n_function_overloads": 1, "n_function_params": 3 }, { "kind": "function", "name": "einops.packing.analyze_pattern", "n_typed": 3, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 3, "n_params": 2, "n_function_overloads": 1, "n_function_params": 2 }, { "kind": "function", "name": "einops.packing.prod", "n_typed": 2, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "attr", "name": "einops.packing.Tensor", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "attr", "name": "einops.packing.Shape", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 } ], "type_ignores": [], "name": "einops.packing", "names": [ "einops.packing.Shape", "einops.packing.Tensor", "einops.packing.analyze_pattern", "einops.packing.pack", "einops.packing.prod", "einops.packing.unpack" ], "n_typable": 13, "n_typed": 13, "n_any": 0, "n_untyped": 0, "n_functions": 4, "n_function_overloads": 4, "n_function_params": 8, "n_methods": 0, "n_method_overloads": 0, "n_method_params": 0, "n_classes": 0, "n_attrs": 2, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/__init__.py", "symbol_reports": [ { "kind": "class", "name": "einops.EinopsError", "methods": [], "properties": [], "attrs": [], "n_typable": 0, "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 0, "n_method_overloads": 0, "n_method_params": 0, "n_attrs": 0, "n_properties": 0 } ], "type_ignores": [], "name": "einops", "names": [ "einops.EinopsError" ], "n_typable": 0, "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 0, "n_method_overloads": 0, "n_method_params": 0, "n_classes": 1, "n_attrs": 0, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/array_api.py", "symbol_reports": [ { "kind": "function", "name": "einops.array_api.reduce", "n_typed": 5, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "einops.array_api.pack", "n_typed": 3, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 3, "n_params": 2, "n_function_overloads": 1, "n_function_params": 2 }, { "kind": "function", "name": "einops.array_api.repeat", "n_typed": 3, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 4, "n_params": 3, "n_function_overloads": 1, "n_function_params": 3 }, { "kind": "function", "name": "einops.array_api.unpack", "n_typed": 4, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 4, "n_params": 3, "n_function_overloads": 1, "n_function_params": 3 }, { "kind": "function", "name": "einops.array_api.rearrange", "n_typed": 3, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 4, "n_params": 3, "n_function_overloads": 1, "n_function_params": 3 }, { "kind": "function", "name": "einops.array_api.asnumpy", "n_typed": 1, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "type_ignores": [], "name": "einops.array_api", "names": [ "einops.array_api.asnumpy", "einops.array_api.pack", "einops.array_api.rearrange", "einops.array_api.reduce", "einops.array_api.repeat", "einops.array_api.unpack" ], "n_typable": 22, "n_typed": 19, "n_any": 0, "n_untyped": 3, "n_functions": 6, "n_function_overloads": 6, "n_function_params": 16, "n_methods": 0, "n_method_overloads": 0, "n_method_params": 0, "n_classes": 0, "n_attrs": 0, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/layers/__init__.py", "symbol_reports": [ { "kind": "attr", "name": "einops.layers.__author__", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "class", "name": "einops.layers.RearrangeMixin", "methods": [ { "kind": "function", "name": "RearrangeMixin.__init__", "n_typed": 2, "n_any": 1, "n_untyped": 0, "n_overloads": 1, "n_typable": 3, "n_params": 2, "n_function_overloads": 1, "n_function_params": 2 }, { "kind": "function", "name": "RearrangeMixin.__repr__", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "RearrangeMixin.multirecipe", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "RearrangeMixin._apply_recipe", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "RearrangeMixin.__getstate__", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "RearrangeMixin.__setstate__", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [ { "kind": "attr", "name": "RearrangeMixin.pattern", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_typable": 1 }, { "kind": "attr", "name": "RearrangeMixin.axes_lengths", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_typable": 1 } ], "n_typable": 12, "n_typed": 4, "n_any": 1, "n_untyped": 7, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 6, "n_method_overloads": 6, "n_method_params": 4, "n_attrs": 2, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.ReduceMixin", "methods": [ { "kind": "function", "name": "ReduceMixin.__init__", "n_typed": 2, "n_any": 1, "n_untyped": 1, "n_overloads": 1, "n_typable": 4, "n_params": 3, "n_function_overloads": 1, "n_function_params": 3 }, { "kind": "function", "name": "ReduceMixin.__repr__", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "ReduceMixin.multirecipe", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "ReduceMixin._apply_recipe", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "ReduceMixin.__getstate__", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "ReduceMixin.__setstate__", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [ { "kind": "attr", "name": "ReduceMixin.pattern", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_typable": 1 }, { "kind": "attr", "name": "ReduceMixin.reduction", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_typable": 1 }, { "kind": "attr", "name": "ReduceMixin.axes_lengths", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_typable": 1 } ], "n_typable": 14, "n_typed": 3, "n_any": 1, "n_untyped": 10, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 6, "n_method_overloads": 6, "n_method_params": 5, "n_attrs": 3, "n_properties": 0 } ], "type_ignores": [], "name": "einops.layers", "names": [ "einops.layers.RearrangeMixin", "einops.layers.ReduceMixin", "einops.layers.__author__" ], "n_typable": 26, "n_typed": 7, "n_any": 2, "n_untyped": 17, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 12, "n_method_overloads": 12, "n_method_params": 9, "n_classes": 2, "n_attrs": 6, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/layers/flax.py", "symbol_reports": [ { "kind": "attr", "name": "einops.layers.flax.__author__", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "class", "name": "einops.layers.flax.Rearrange", "methods": [ { "kind": "function", "name": "Rearrange.setup", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "Rearrange.__call__", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [ { "kind": "attr", "name": "Rearrange.pattern", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "Rearrange.sizes", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 } ], "n_typable": 5, "n_typed": 2, "n_any": 0, "n_untyped": 3, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 2, "n_method_overloads": 2, "n_method_params": 1, "n_attrs": 2, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.flax.EinMix", "methods": [ { "kind": "function", "name": "EinMix.setup", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "EinMix._create_parameters", "n_typed": 0, "n_any": 0, "n_untyped": 5, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix._create_rearrange_layers", "n_typed": 4, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix.__call__", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [ { "kind": "attr", "name": "EinMix.pattern", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "EinMix.weight_shape", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "EinMix.bias_shape", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "EinMix.sizes", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 } ], "n_typable": 17, "n_typed": 8, "n_any": 0, "n_untyped": 9, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 4, "n_method_overloads": 4, "n_method_params": 9, "n_attrs": 4, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.flax.Reduce", "methods": [ { "kind": "function", "name": "Reduce.setup", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "Reduce.__call__", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [ { "kind": "attr", "name": "Reduce.pattern", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "Reduce.reduction", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "Reduce.sizes", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 } ], "n_typable": 6, "n_typed": 3, "n_any": 0, "n_untyped": 3, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 2, "n_method_overloads": 2, "n_method_params": 1, "n_attrs": 3, "n_properties": 0 } ], "type_ignores": [], "name": "einops.layers.flax", "names": [ "einops.layers.flax.EinMix", "einops.layers.flax.Rearrange", "einops.layers.flax.Reduce", "einops.layers.flax.__author__" ], "n_typable": 28, "n_typed": 13, "n_any": 0, "n_untyped": 15, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 8, "n_method_overloads": 8, "n_method_params": 11, "n_classes": 3, "n_attrs": 10, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/layers/keras.py", "symbol_reports": [ { "kind": "attr", "name": "einops.layers.keras.keras_custom_objects", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "attr", "name": "einops.layers.keras.__author__", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 } ], "type_ignores": [], "name": "einops.layers.keras", "names": [ "einops.layers.keras.__author__", "einops.layers.keras.keras_custom_objects" ], "n_typable": 0, "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 0, "n_method_overloads": 0, "n_method_params": 0, "n_classes": 0, "n_attrs": 2, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/layers/oneflow.py", "symbol_reports": [ { "kind": "attr", "name": "einops.layers.oneflow.__author__", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "class", "name": "einops.layers.oneflow.Rearrange", "methods": [ { "kind": "function", "name": "Rearrange.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 2, "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 1, "n_method_overloads": 1, "n_method_params": 1, "n_attrs": 0, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.oneflow.EinMix", "methods": [ { "kind": "function", "name": "EinMix._create_parameters", "n_typed": 0, "n_any": 0, "n_untyped": 5, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix._create_rearrange_layers", "n_typed": 4, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 12, "n_typed": 4, "n_any": 0, "n_untyped": 8, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 3, "n_method_overloads": 3, "n_method_params": 9, "n_attrs": 0, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.oneflow.Reduce", "methods": [ { "kind": "function", "name": "Reduce.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 2, "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 1, "n_method_overloads": 1, "n_method_params": 1, "n_attrs": 0, "n_properties": 0 } ], "type_ignores": [], "name": "einops.layers.oneflow", "names": [ "einops.layers.oneflow.EinMix", "einops.layers.oneflow.Rearrange", "einops.layers.oneflow.Reduce", "einops.layers.oneflow.__author__" ], "n_typable": 16, "n_typed": 4, "n_any": 0, "n_untyped": 12, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 5, "n_method_overloads": 5, "n_method_params": 11, "n_classes": 3, "n_attrs": 1, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/layers/paddle.py", "symbol_reports": [ { "kind": "attr", "name": "einops.layers.paddle.__author__", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "class", "name": "einops.layers.paddle.Rearrange", "methods": [ { "kind": "function", "name": "Rearrange.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 2, "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 1, "n_method_overloads": 1, "n_method_params": 1, "n_attrs": 0, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.paddle.EinMix", "methods": [ { "kind": "function", "name": "EinMix._create_parameters", "n_typed": 0, "n_any": 0, "n_untyped": 5, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix._create_rearrange_layers", "n_typed": 4, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 12, "n_typed": 4, "n_any": 0, "n_untyped": 8, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 3, "n_method_overloads": 3, "n_method_params": 9, "n_attrs": 0, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.paddle.Reduce", "methods": [ { "kind": "function", "name": "Reduce.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 2, "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 1, "n_method_overloads": 1, "n_method_params": 1, "n_attrs": 0, "n_properties": 0 } ], "type_ignores": [], "name": "einops.layers.paddle", "names": [ "einops.layers.paddle.EinMix", "einops.layers.paddle.Rearrange", "einops.layers.paddle.Reduce", "einops.layers.paddle.__author__" ], "n_typable": 16, "n_typed": 4, "n_any": 0, "n_untyped": 12, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 5, "n_method_overloads": 5, "n_method_params": 11, "n_classes": 3, "n_attrs": 1, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/layers/tensorflow.py", "symbol_reports": [ { "kind": "attr", "name": "einops.layers.tensorflow.__author__", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "class", "name": "einops.layers.tensorflow.Rearrange", "methods": [ { "kind": "function", "name": "Rearrange.build", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "Rearrange.call", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "Rearrange.get_config", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 } ], "properties": [], "attrs": [], "n_typable": 5, "n_typed": 0, "n_any": 0, "n_untyped": 5, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 3, "n_method_overloads": 3, "n_method_params": 2, "n_attrs": 0, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.tensorflow.EinMix", "methods": [ { "kind": "function", "name": "EinMix._create_parameters", "n_typed": 0, "n_any": 0, "n_untyped": 5, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix._create_rearrange_layers", "n_typed": 4, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix.build", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "EinMix.call", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "EinMix.get_config", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 } ], "properties": [], "attrs": [], "n_typable": 15, "n_typed": 4, "n_any": 0, "n_untyped": 11, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 5, "n_method_overloads": 5, "n_method_params": 10, "n_attrs": 0, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.tensorflow.Reduce", "methods": [ { "kind": "function", "name": "Reduce.build", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "Reduce.call", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "Reduce.get_config", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 } ], "properties": [], "attrs": [], "n_typable": 5, "n_typed": 0, "n_any": 0, "n_untyped": 5, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 3, "n_method_overloads": 3, "n_method_params": 2, "n_attrs": 0, "n_properties": 0 } ], "type_ignores": [], "name": "einops.layers.tensorflow", "names": [ "einops.layers.tensorflow.EinMix", "einops.layers.tensorflow.Rearrange", "einops.layers.tensorflow.Reduce", "einops.layers.tensorflow.__author__" ], "n_typable": 25, "n_typed": 4, "n_any": 0, "n_untyped": 21, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 11, "n_method_overloads": 11, "n_method_params": 14, "n_classes": 3, "n_attrs": 1, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/layers/torch.py", "symbol_reports": [ { "kind": "attr", "name": "einops.layers.torch.__author__", "n_typed": 0, "n_any": 0, "n_untyped": 0, "n_typable": 0 }, { "kind": "class", "name": "einops.layers.torch.Rearrange", "methods": [ { "kind": "function", "name": "Rearrange.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "Rearrange._apply_recipe", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 4, "n_typed": 0, "n_any": 0, "n_untyped": 4, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 2, "n_method_overloads": 2, "n_method_params": 2, "n_attrs": 0, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.torch.EinMix", "methods": [ { "kind": "function", "name": "EinMix._create_parameters", "n_typed": 0, "n_any": 0, "n_untyped": 5, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix._create_rearrange_layers", "n_typed": 4, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 5, "n_params": 4, "n_function_overloads": 1, "n_function_params": 4 }, { "kind": "function", "name": "EinMix.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 12, "n_typed": 4, "n_any": 0, "n_untyped": 8, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 3, "n_method_overloads": 3, "n_method_params": 9, "n_attrs": 0, "n_properties": 0 }, { "kind": "class", "name": "einops.layers.torch.Reduce", "methods": [ { "kind": "function", "name": "Reduce.forward", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "Reduce._apply_recipe", "n_typed": 0, "n_any": 0, "n_untyped": 2, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [], "n_typable": 4, "n_typed": 0, "n_any": 0, "n_untyped": 4, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 2, "n_method_overloads": 2, "n_method_params": 2, "n_attrs": 0, "n_properties": 0 } ], "type_ignores": [], "name": "einops.layers.torch", "names": [ "einops.layers.torch.EinMix", "einops.layers.torch.Rearrange", "einops.layers.torch.Reduce", "einops.layers.torch.__author__" ], "n_typable": 20, "n_typed": 4, "n_any": 0, "n_untyped": 16, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 7, "n_method_overloads": 7, "n_method_params": 13, "n_classes": 3, "n_attrs": 1, "n_properties": 0, "n_type_ignores": 0 }, { "path": "einops/parsing.py", "symbol_reports": [ { "kind": "class", "name": "einops.parsing.ParsedExpression", "methods": [ { "kind": "function", "name": "ParsedExpression.__init__", "n_typed": 3, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 4, "n_params": 3, "n_function_overloads": 1, "n_function_params": 3 }, { "kind": "function", "name": "ParsedExpression.flat_axes_order", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "ParsedExpression.has_composed_axes", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 }, { "kind": "function", "name": "ParsedExpression.check_axis_name_return_reason", "n_typed": 3, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 3, "n_params": 2, "n_function_overloads": 1, "n_function_params": 2 }, { "kind": "function", "name": "ParsedExpression.check_axis_name", "n_typed": 2, "n_any": 0, "n_untyped": 0, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 } ], "properties": [], "attrs": [ { "kind": "attr", "name": "ParsedExpression.has_ellipsis", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "ParsedExpression.has_ellipsis_parenthesized", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "ParsedExpression.identifiers", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "ParsedExpression.has_non_unitary_anonymous_axes", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 }, { "kind": "attr", "name": "ParsedExpression.composition", "n_typed": 1, "n_any": 0, "n_untyped": 0, "n_typable": 1 } ], "n_typable": 16, "n_typed": 15, "n_any": 0, "n_untyped": 1, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 5, "n_method_overloads": 5, "n_method_params": 6, "n_attrs": 5, "n_properties": 0 }, { "kind": "class", "name": "einops.parsing.AnonymousAxis", "methods": [ { "kind": "function", "name": "AnonymousAxis.__init__", "n_typed": 1, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 2, "n_params": 1, "n_function_overloads": 1, "n_function_params": 1 }, { "kind": "function", "name": "AnonymousAxis.__repr__", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_overloads": 1, "n_typable": 1, "n_params": 0, "n_function_overloads": 1, "n_function_params": 0 } ], "properties": [], "attrs": [ { "kind": "attr", "name": "AnonymousAxis.value", "n_typed": 0, "n_any": 0, "n_untyped": 1, "n_typable": 1 } ], "n_typable": 4, "n_typed": 1, "n_any": 0, "n_untyped": 3, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 2, "n_method_overloads": 2, "n_method_params": 1, "n_attrs": 1, "n_properties": 0 } ], "type_ignores": [], "name": "einops.parsing", "names": [ "einops.parsing.AnonymousAxis", "einops.parsing.ParsedExpression" ], "n_typable": 20, "n_typed": 16, "n_any": 0, "n_untyped": 4, "n_functions": 0, "n_function_overloads": 0, "n_function_params": 0, "n_methods": 7, "n_method_overloads": 7, "n_method_params": 7, "n_classes": 2, "n_attrs": 6, "n_properties": 0, "n_type_ignores": 0 } ], "typecheckers": {}, "n_modules": 12, "n_typable": 230, "n_typed": 123, "n_any": 6, "n_untyped": 101, "n_functions": 16, "n_function_overloads": 22, "n_function_params": 42, "n_methods": 56, "n_method_params": 83, "n_method_overloads": 56, "n_classes": 21, "n_attrs": 43, "n_properties": 0, "type_ignores": [], "n_type_ignores": 0 }