mostly stylistic refactor + improved edge case handling for softmax, log_softmax#3328

jachymb wants to merge 3 commits into

stan-dev:developstan-dev/math:developfrom

jachymb:jachymb_softmax_refactorjachymb/math:jachymb_softmax_refactorCopy head branch name to clipboard