Skip to content

Commit

Permalink
add TODO
Browse files Browse the repository at this point in the history
  • Loading branch information
Patrick-Star125 committed Jun 8, 2023
1 parent 79c5884 commit 4e4665d
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions python/paddle/sparse/unary.py
Original file line number Diff line number Diff line change
Expand Up @@ -1111,6 +1111,8 @@ def svd_lowrank(x, q=6, niter=2, M=None):

if len(x.shape) != 2:
raise ValueError('input is expected to be 2-dimensional tensor')
# TODO: complement sparse_csr_tensor test
# when sparse.sum with axis(-2) is implemented
s_sum = paddle.sparse.sum(x, axis=-2)
s_val = s_sum.values() / m
c = paddle.sparse.sparse_coo_tensor(
Expand Down

0 comments on commit 4e4665d

Please sign in to comment.