We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
_compute_normalized_gramian
1 parent d5a36bf commit 0a6244fCopy full SHA for 0a6244f
src/torchjd/aggregation/_gramian_utils.py
@@ -10,11 +10,6 @@ def _compute_gramian(matrix: Tensor) -> Tensor:
10
return matrix @ matrix.T
11
12
13
-def _compute_normalized_gramian(matrix: Tensor, eps: float) -> Tensor:
14
- gramian = _compute_gramian(matrix)
15
- return _normalize(gramian, eps)
16
-
17
18
def _normalize(gramian: Tensor, eps: float) -> Tensor:
19
"""
20
Normalizes the gramian with respect to the Frobenius norm.
0 commit comments