Apply_along_axis和Dask Arrays

Question

使用apply_along_axis方法后，我对Dask数组的形状有疑问：

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)。