Apply_along_axis и Dask Arrays

Question

У меня есть вопрос о форме массива Dask после использования метода apply_along_axis:

import dask.array as da
array = da.from_array(np.array([[1,2,3,4], [5,6,7,8], \
                                [9, 10, 11, 12], [13, 14, 15, 16]]), \
                      chunks=(2,2))   

Теперь, если

a = da.apply_along_axis((lambda x: x/sum(x)), axis=0, arr=array)

, то

a.shape == (1,4)

, но

a.compute().shape == (4,4)

Это правильно?Я думаю, что a.shape также должно быть (4,4).