# {py:mod}`evox.utils.parameters_and_vector`

```{py:module} evox.utils.parameters_and_vector
```

```{autodoc2-docstring} evox.utils.parameters_and_vector
:allowtitles:
```

## Module Contents

### Classes

````{list-table}
:class: autosummary longtable
:align: left

* - {py:obj}`ParamsAndVector <evox.utils.parameters_and_vector.ParamsAndVector>`
  - ```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector
    :summary:
    ```
````

### API

`````{py:class} ParamsAndVector(dummy_model: torch.nn.Module)
:canonical: evox.utils.parameters_and_vector.ParamsAndVector

Bases: {py:obj}`evox.core.ModuleBase`

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector
```

```{rubric} Initialization
```

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector.__init__
```

````{py:method} _tree_flatten(x: typing.Dict[str, torch.nn.Parameter]) -> typing.List[torch.nn.Parameter]
:canonical: evox.utils.parameters_and_vector.ParamsAndVector._tree_flatten

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector._tree_flatten
```

````

````{py:method} _tree_unflatten(x: typing.List[torch.nn.Parameter]) -> typing.Dict[str, torch.nn.Parameter]
:canonical: evox.utils.parameters_and_vector.ParamsAndVector._tree_unflatten

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector._tree_unflatten
```

````

````{py:method} to_vector(params: typing.Dict[str, torch.nn.Parameter]) -> torch.Tensor
:canonical: evox.utils.parameters_and_vector.ParamsAndVector.to_vector

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector.to_vector
```

````

````{py:method} batched_to_vector(batched_params: typing.Dict[str, torch.nn.Parameter]) -> torch.Tensor
:canonical: evox.utils.parameters_and_vector.ParamsAndVector.batched_to_vector

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector.batched_to_vector
```

````

````{py:method} to_params(vector: torch.Tensor) -> typing.Dict[str, torch.nn.Parameter]
:canonical: evox.utils.parameters_and_vector.ParamsAndVector.to_params

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector.to_params
```

````

````{py:method} batched_to_params(vectors: torch.Tensor) -> typing.Dict[str, torch.nn.Parameter]
:canonical: evox.utils.parameters_and_vector.ParamsAndVector.batched_to_params

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector.batched_to_params
```

````

````{py:method} forward(x: torch.Tensor) -> typing.Dict[str, torch.nn.Parameter]
:canonical: evox.utils.parameters_and_vector.ParamsAndVector.forward

```{autodoc2-docstring} evox.utils.parameters_and_vector.ParamsAndVector.forward
```

````

`````