refine example code

PaddlePaddle · Jun 7, 2023 · 79c5884 · 79c5884
1 parent e3ca1d5
commit 79c5884
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/sparse/unary.py b/python/paddle/sparse/unary.py
@@ -965,7 +965,8 @@ def pca_lowrank(x, q=None, center=True, niter=2, name=None):
 
  print("sparse.pca_lowrank API only support CUDA 11.x")
  U, S, V = None, None, None
- # U, S, V = pca_lowrank(sparse_x)
+ # use code blow when your device CUDA version >= 11.0
+ # U, S, V = paddle.sparse.pca_lowrank(sparse_x)
 
  print(U)
  # Tensor(shape=[5, 5], dtype=float64, place=Place(gpu:0), stop_gradient=True,