fixed the softmax constructor arguments order.