sqrt¶

paddle.sparse. sqrt ( x, name=None ) [source]

Calculate elementwise sqrt of SparseTensor, requiring x to be a SparseCooTensor or SparseCsrTensor.

\[out = sqrt(x)\]

Parameters

• x (Tensor) – The input Sparse Tensor with data type float32, float64.

• name (str, optional) – Name for the operation (optional, default is None). For more information, please refer to Name.

Returns

A Sparse Tensor with the same data type and shape as x .

Examples

import paddle

dense_x = paddle.to_tensor([-2., 0., 1.])
sparse_x = dense_x.to_sparse_coo(1)
out = paddle.sparse.sqrt(sparse_x)