relu

paddle.sparse.nn.functional. relu ( x, name=None ) [source]

sparse relu activation, requiring x to be a SparseCooTensor or SparseCsrTensor.

\[out = max(x, 0)\]
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.nn.functional.relu(sparse_x)
>>> print(out)
Tensor(shape=[3], dtype=paddle.float32, place=Place(cpu), stop_gradient=True,
       indices=[[0, 2]],
       values=[0., 1.])