docstring parameter fixed

Author: mkumar73

mambular/arch_utils/layer_utils/sparsemax.py

@@ -92,7 +92,7 @@ def _threshold_and_support(input_, dim=-1):
 
         Parameters
         ----------
-        input : torch.Tensor
+        input_ : torch.Tensor
             The input tensor on which to compute the threshold and support.
         dim : int, optional
             Dimension along which to compute the threshold and support. Default is -1.