稀疏 Softmax 激活层,创建一个可调用对象以计算输入 x 的 Softmax 。
当输入 x 为 SparseCsrTensor 时,仅支持 axis=-1,是由于 Csr 稀疏存储格式,更适合按行读取数据。
如果将 x 从稀疏矩阵转换为稠密矩阵, i 代表行数, j 代表列数,且 axis=-1 时有如下公式:
- axis (int, 可选) - 指定对输入 SparseTensor 计算 softmax 的轴。对于 SparseCsrTensor,仅支持 axis=-1。默认值:-1。
- name (str,可选) - 具体用法请参见
api_guide_Name
,一般无需设置,默认值为 None。
- input:任意形状的 SparseTensor。
- output:和 input 具有相同形状和数据类型的 SparseTensor。
COPY-FROM: paddle.sparse.nn.Softmax