add spatial_argmax
fix test for criterion on GPU
remove manualSeed from test