IVGCVSW-836 Remove 2D validation in Softmax