[ torch 参数更多 ] torch.sparse.softmax¶

torch.sparse.softmax ¶

          torch.sparse.softmax(input, dim, dtype=None)

         

          paddle.sparse.nn.functional.softmax(x, axis=-1, name=None)

         

Pytorch 相比 Paddle 支持更多其他参数，具体如下：

PyTorch	PaddlePaddle	备注
input	x	输入的稀疏 Tensor，仅参数名不一致。
dim	axis	指定对输入 SparseTensor 计算 softmax 的轴，Paddle 的默认值：-1。仅参数名不一致。
dtype	-	指定数据类型，可选项，Pytorch 默认值为 None，Paddle 无此参数，需要转写。
### 转写示例
#### dytpe：指定数据类型
```Python
# Pytorch 写法
y = torch.sparse.softmax(x, dim=-1, dtype=torch.float32)

y = paddle.sparse.cast(x, value_dtype=’float32’) y = paddle.sparse.nn.functional.softmax(y)