returnn 1.20240605.145641__tar.gz → 1.20240606.94409__tar.gz

{returnn-1.20240605.145641/returnn.egg-info → returnn-1.20240606.94409}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: returnn
-Version: 1.20240605.145641
+Version: 1.20240606.94409
 Summary: The RWTH extensible training framework for universal recurrent neural networks
 Home-page: https://github.com/rwth-i6/returnn/
 Author: Albert Zeyer

returnn-1.20240606.94409/_setup_info_generated.py

@@ -0,0 +1,2 @@
+version = '1.20240606.094409'
+long_version = '1.20240606.094409+git.b567b92'

{returnn-1.20240605.145641 → returnn-1.20240606.94409}/returnn/frontend/parameter.py

@@ -24,9 +24,10 @@ class Parameter(Tensor[T]):
 
     def __init__(
         self,
-        dims: Sequence[Dim],
+        dims_or_tensor: Union[None, Sequence[Dim], Tensor] = None,
         dtype: Optional[str] = None,
         *,
+        dims: Optional[Sequence[Dim]] = None,
         sparse_dim: Optional[Dim] = None,
         trainable: Optional[bool] = None,
         auxiliary: bool = False,
@@ -37,6 +38,7 @@ class Parameter(Tensor[T]):
         device: Optional[str] = None,
     ):
         """
+        :param dims_or_tensor:
         :param dims:
         :param dtype:
         :param sparse_dim:
@@ -52,12 +54,31 @@ class Parameter(Tensor[T]):
         :param raw_tensor:
         :param device:
         """
+        if dims_or_tensor is None:
+            if dims is None:
+                raise ValueError("rf.Parameter: dims must be set.")
+        elif isinstance(dims_or_tensor, (tuple, list)):
+            if dims is not None:
+                raise ValueError("rf.Parameter: dims is set twice.")
+            dims = dims_or_tensor
+        elif isinstance(dims_or_tensor, Tensor):
+            if initial is not None:
+                raise ValueError("rf.Parameter: initial is set twice.")
+            initial = dims_or_tensor
+            if dims is None:
+                dims = initial.dims
+            if dtype is None:
+                dtype = initial.dtype
+            if sparse_dim is None:
+                sparse_dim = initial.sparse_dim
+        else:
+            raise TypeError(f"rf.Parameter: invalid type for dims_or_tensor: {type(dims_or_tensor)}")
         if not all(isinstance(dim, Dim) for dim in dims):
-            raise TypeError(f"shape {dims} must be a sequence of Dim")
+            raise TypeError(f"rf.Parameter: shape {dims} must be a sequence of Dim")
         if not all(isinstance(dim.dimension, int) for dim in dims):
-            raise ValueError(f"shape {dims} must be static")
+            raise ValueError(f"rf.Parameter: shape {dims} must be static")
         if len(dims) != len(set((d, d.match_priority) for d in dims)):
-            raise ValueError(f"shape {dims} dims must be unique")
+            raise ValueError(f"rf.Parameter: shape {dims} dims must be unique")
         super(Parameter, self).__init__(
             "parameter",
             dims=dims,

{returnn-1.20240605.145641 → returnn-1.20240606.94409/returnn.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: returnn
-Version: 1.20240605.145641
+Version: 1.20240606.94409
 Summary: The RWTH extensible training framework for universal recurrent neural networks
 Home-page: https://github.com/rwth-i6/returnn/
 Author: Albert Zeyer

returnn-1.20240605.145641/_setup_info_generated.py

@@ -1,2 +0,0 @@
-version = '1.20240605.145641'
-long_version = '1.20240605.145641+git.24cc84b'