fix nan mean#18

Jah-yee wants to merge 3 commits into

thinking-machines-lab:mainthinking-machines-lab/batch_invariant_ops:mainfrom

Jah-yee:fix-nan-meanJah-yee/batch_invariant_ops:fix-nan-meanCopy head branch name to clipboard