fix SpatialSoftMax bug and add unit tests
1 file changed