The SoftArgmax for 2D and 3D by implemented in PyTorch. install pip install pipenv pipenv install test pipenv install --dev pytest
