Skip to content

Commit 0a6244f

Browse files
committed
Remove _compute_normalized_gramian
1 parent d5a36bf commit 0a6244f

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

src/torchjd/aggregation/_gramian_utils.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,6 @@ def _compute_gramian(matrix: Tensor) -> Tensor:
1010
return matrix @ matrix.T
1111

1212

13-
def _compute_normalized_gramian(matrix: Tensor, eps: float) -> Tensor:
14-
gramian = _compute_gramian(matrix)
15-
return _normalize(gramian, eps)
16-
17-
1813
def _normalize(gramian: Tensor, eps: float) -> Tensor:
1914
"""
2015
Normalizes the gramian with respect to the Frobenius norm.

0 commit comments

Comments
 (0)