fix logsumexp and softmax for dim >= 3