Skip to content

Commit ada66ad

Browse files
Update src/torchjd/scalarization/README.md
Co-authored-by: Pierre Quinton <pierre.quinton@gmail.com>
1 parent 0d342cb commit ada66ad

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/torchjd/scalarization/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@ def forward(self, values: Tensor, /) -> Tensor:
1616
...
1717
```
1818

19-
- **Any shape in, scalar out:** it reduces over *all* elements of `values` (0-dim, vector, matrix,
20-
higher-dim) into a 0-dim scalar.
19+
- **Any shape in, scalar out:** it reduces over *all* dimensions of `values` (scalar, vector, matrix,
20+
etc...) into a scalar.
2121
- **`values`, not `losses`:** a scalarizer is generic and not tied to losses.
2222
- **Pure and differentiable:** the output depends only on `values` and the configured parameters, so
2323
that `scalarizer(values).backward()` produces the gradient.

0 commit comments

Comments
 (0)