mean along dims (tested)