/PyTorch-Soft-Argmax

PyTorch implementation of Soft-Argmax in 1D/2D/3D

Primary LanguageJupyter Notebook

Stargazers