Add default_list_dim to wrap()

tum-pbs · Feb 1, 2024 · b0f4a7c · b0f4a7c
1 parent e580b87
commit b0f4a7c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/phiml/math/_tensors.py b/phiml/math/_tensors.py
@@ -1669,10 +1669,9 @@ def tensor(data,
         raise ValueError(f"{type(data)} is not supported. Only (Tensor, tuple, list, np.ndarray, native tensors) are allowed.\nCurrent backends: {BACKENDS}")
 
 
-def wrap(data,
-         *shape: Shape) -> Tensor:
+def wrap(data, *shape: Shape, default_list_dim=channel('vector')) -> Tensor:
     """ Short for `phiml.math.tensor()` with `convert=False`. """
-    return tensor(data, *shape, convert=False)  # TODO inline, simplify
+    return tensor(data, *shape, convert=False, default_list_dim=default_list_dim)
 
 
 def layout(objects, *shape: Shape) -> Tensor: