Update softmax implementation with dims argument#90

Merged

DhairyaLGandhi merged 3 commits intomasterfrom ap/softmax_fix

Jan 3, 2021

+10-8

Commits on Dec 21, 2020

Fix softmax implementation
avik-pal
committed

Commits on Dec 23, 2020

Update src/lib/array.jl

DhairyaLGandhi
and
devmotion
authored

Commits on Jan 3, 2021

Update tests to use dims argument
avik-pal
committed