torch-einops-utils 0.0.21__py3-none-any.whl → 0.0.23__py3-none-any.whl

torch_einops_utils/torch_einops_utils.py

@@ -244,6 +244,7 @@ def pad_sequence(
     value = 0.,
     left = False,
     dim_stack = 0,
+    return_stacked = True,
     return_lens = False,
     pad_lens = False # returns padding length instead of sequence lengths
 ):
@@ -258,15 +259,17 @@ def pad_sequence(
     pad_fn = pad_left_at_dim if left else pad_right_at_dim
     padded_tensors = [pad_fn(t, max_len - t_len, dim = dim, value = value) for t, t_len in zip(tensors, lens)]
 
-    stacked = stack(padded_tensors, dim = dim_stack)
+    output = padded_tensors
+    if return_stacked:
+        output = stack(output, dim = dim_stack)
 
     if not return_lens:
-        return stacked
+        return output
 
     if pad_lens:
         lens = max_len - lens
 
-    return stacked, lens
+    return output, lens
 
 # tree flatten with inverse
 

{torch_einops_utils-0.0.21.dist-info → torch_einops_utils-0.0.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: torch-einops-utils
-Version: 0.0.21
+Version: 0.0.23
 Summary: Personal utility functions
 Project-URL: Homepage, https://pypi.org/project/torch-einops-utils/
 Project-URL: Repository, https://github.com/lucidrains/torch-einops-utils

torch_einops_utils-0.0.23.dist-info/RECORD

@@ -0,0 +1,7 @@
+torch_einops_utils/__init__.py,sha256=l7S9xJOlbMfwofQD3RI2Z4QxT3H713XSPc1zN_kF4Ww,982
+torch_einops_utils/save_load.py,sha256=K-i7nmLyXBHdAfBLN3rGQzI3NVf6RRwF_GcrKQnfQsc,2669
+torch_einops_utils/torch_einops_utils.py,sha256=raKFiWpRgGE7Vf7tTqkokU9525gr54lJMpQy_jnUTVI,6498
+torch_einops_utils-0.0.23.dist-info/METADATA,sha256=vN8uy5uFGwLhsIqkEXwf8uvzvmpVH9Vm2gprQjrzKqw,2139
+torch_einops_utils-0.0.23.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+torch_einops_utils-0.0.23.dist-info/licenses/LICENSE,sha256=e6AOF7Z8EFdK3IdcL0x0fLw4cY7Q0d0kNR0o0TmBewM,1066
+torch_einops_utils-0.0.23.dist-info/RECORD,,

torch_einops_utils-0.0.21.dist-info/RECORD

@@ -1,7 +0,0 @@
-torch_einops_utils/__init__.py,sha256=l7S9xJOlbMfwofQD3RI2Z4QxT3H713XSPc1zN_kF4Ww,982
-torch_einops_utils/save_load.py,sha256=K-i7nmLyXBHdAfBLN3rGQzI3NVf6RRwF_GcrKQnfQsc,2669
-torch_einops_utils/torch_einops_utils.py,sha256=exfuRHRBP1YK_3BIpjquFm3fJIX1d9-rNYP1wt9lzoE,6427
-torch_einops_utils-0.0.21.dist-info/METADATA,sha256=1NMFNMuPjmI5xIw4d1r46XtEcotl-OjRsjk8p6lh68k,2139
-torch_einops_utils-0.0.21.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-torch_einops_utils-0.0.21.dist-info/licenses/LICENSE,sha256=e6AOF7Z8EFdK3IdcL0x0fLw4cY7Q0d0kNR0o0TmBewM,1066
-torch_einops_utils-0.0.21.dist-info/RECORD,,