add downsample_along_axis tool to the downsample code