[add-fire] Fix masked mean for 2d tensors#4364

ervteng merged 2 commits intodevelop-add-fireUnity-Technologies/ml-agents:develop-add-firefrom

develop-add-fire-continuousUnity-Technologies/ml-agents:develop-add-fire-continuousCopy head branch name to clipboard