经过比较，pose-estimation/stitched/soft_argmax.py at master · MWPainter/pose-estimation · GitHub表现好，和torch.argmax结果保持一致

GitHub - Fdevmsy/PyTorch-Soft-Argmax: PyTorch implementation of Soft-Argmax in 1D/2D/3D这个会出现错误计算结果，没有第一个好