Raise error when inputs and mask don't have same shape in Softmax

Raise error when inputs and mask don't have same shape in Softmax #978