Разделение Numpy-массива на части при помощи np.array_split

Обрабатываю массив arr, который является изображением, преобразованным в numpy массив. Он трёхмерен, первые два вложенных элемента - строки\столбцы, 3 - пиксели в строках\столбцах. Я пытаюсь разделить его при помощи np.array-split(), я создал такой код:

import numpy as np
f = np.array_split(arr,16)
print(f)
for i in f:
    f[i]= np.array_split(f[i],16)

Однако получаю следующую ошибку:

TypeError: only integer scalar arrays can be converted to a scalar index