relu¶

paddle.sparse.nn.functional. relu ( x, name=None ) [源代码] ¶

稀疏 relu 激活函数，要求输入 x 为 SparseCooTensor 或 SparseCsrTensor 。

\[relu(x) = max(x, 0)\]

其中，\(x\) 为输入的 Tensor。

参数¶

x (Tensor) - 输入的稀疏 Tensor，可以是 SparseCooTensor 或 SparseCsrTensor，数据类型为 float32、float64。

name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回¶

多维稀疏 Tensor, 数据类型和稀疏格式与 x 相同。

代码示例¶

          >>> 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.])

         