Merge pull request #147 from jeffin07/softmax

softmax bug fix

Author: imdeepmind

neuralpy/layers/activation_functions/softmax.py

@@ -29,7 +29,7 @@ def __init__(self, dim=None, name=None):
         """
         super().__init__(_Softmax, "Softmax", layer_name=name)
 
-        if not isinstance(dim, int):
+        if dim is not None and not isinstance(dim, int):
             raise ValueError("Please provide a valid dim")
 
         self.__dim = dim