[unittest] specify softmax template type