/sparsemax

Primary LanguagePython

参考 https://github.com/aced125/sparsemax 实现了新版本的sparsemax函数

使用方法参考原仓库(使用本仓库版本的sparsemax不能直接用pip安装,而是手动下载sparsemax.py文件,然后添加到工程中)

1)修复了原来仓库的sparsemax不支持1D tensor的bug

2)修复了原来仓库的sparsemax在2D tensor输入情况下dim=0的bug