Как использовать функцию накопления Python с 3D-массивом

Недавно я попытался перенести некоторые функции из Matlab в Python. Функция в Matlab, называемая accumarray, accumarray(3d_array(10*10*10),1), на выходе также представляет собой массив 101010 3d.

Но когда я пытаюсь сделать то же самое в функции accum в python, я не могу напрямую написать 1 как vals в accum(3d_array(101010), vals), знаете ли вы, как я могу заставить эту функцию работать, а также выводить 3D-массив?

Спасибо вам большое за ваше время!