torch-einops-utils 0.0.4__py3-none-any.whl

torch_einops_utils/__init__.py

@@ -0,0 +1,7 @@
+
+from torch_einops_utils.torch_einops_utils import (
+    pad_at_dim,
+    pad_left_at_dim,
+    pad_right_at_dim,
+    pack_with_inverse
+)

torch_einops_utils/torch_einops_utils.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+import torch
+from torch import is_tensor
+import torch.nn.functional as F
+
+from einops import rearrange, repeat, reduce, pack, unpack
+
+# helper functions
+
+def exists(v):
+    return v is not None
+
+def default(v, d):
+    return v if exists(v) else d
+
+def first(arr):
+    return arr[0]
+
+# exported functions
+
+# padding
+
+def pad_at_dim(
+    t,
+    pad: tuple[int, int],
+    *,
+    dim = -1,
+    value = 0.
+):
+    dims_from_right = (- dim - 1) if dim < 0 else (t.ndim - dim - 1)
+    zeros = ((0, 0) * dims_from_right)
+    return F.pad(t, (*zeros, *pad), value = value)
+
+def pad_left_at_dim(t, pad: int, **kwargs):
+    return pad_at_dim(t, (pad, 0), **kwargs)
+
+def pad_right_at_dim(t, pad: int, **kwargs):
+    return pad_at_dim(t, (0, pad), **kwargs)
+
+# einops pack
+
+def pack_with_inverse(t, pattern):
+    is_one = is_tensor(t)
+
+    if is_one:
+        t = [t]
+
+    packed, packed_shape = pack(t, pattern)
+
+    def inverse(out, inv_pattern = None):
+        inv_pattern = default(inv_pattern, pattern)
+        out = unpack(out, packed_shape, inv_pattern)
+
+        if is_one:
+            out = first(out)
+
+        return out
+
+    return packed, inverse

torch_einops_utils-0.0.4.dist-info/METADATA

@@ -0,0 +1,46 @@
+Metadata-Version: 2.4
+Name: torch-einops-utils
+Version: 0.0.4
+Summary: Personal utility functions
+Project-URL: Homepage, https://pypi.org/project/torch-einops-utils/
+Project-URL: Repository, https://github.com/lucidrains/torch-einops-utils
+Author-email: Phil Wang <lucidrains@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2026 Phil Wang
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+License-File: LICENSE
+Keywords: einops,torch
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.9
+Requires-Dist: einops>=0.8.1
+Requires-Dist: torch>=2.5
+Provides-Extra: examples
+Provides-Extra: test
+Requires-Dist: pytest; extra == 'test'
+Description-Content-Type: text/markdown
+
+## torch-einops-utils
+
+Some utility functions to help myself (and perhaps others) to go faster with ML/AI work

torch_einops_utils-0.0.4.dist-info/RECORD

@@ -0,0 +1,6 @@
+torch_einops_utils/__init__.py,sha256=-8iTNDGOPUv59HPx-Jm-4RSkTZe9WEbtRA_Hxc7aw7U,136
+torch_einops_utils/torch_einops_utils.py,sha256=2p-gGpYXQojtxSXsLyRES3dyjjnHB6qrF6FU8BnQ5M0,1179
+torch_einops_utils-0.0.4.dist-info/METADATA,sha256=UXnpWSU1lX0iZuWXE3_NRouJSyilKkVN_NPC14sSqrA,2141
+torch_einops_utils-0.0.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+torch_einops_utils-0.0.4.dist-info/licenses/LICENSE,sha256=e6AOF7Z8EFdK3IdcL0x0fLw4cY7Q0d0kNR0o0TmBewM,1066
+torch_einops_utils-0.0.4.dist-info/RECORD,,

torch_einops_utils-0.0.4.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

torch_einops_utils-0.0.4.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Phil Wang
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.