I am not sure on 'axis=' either -- so far we did not have it too often in our API and had enough of ```_sample``` and ```_feature``` functions.  So probably Ok

BUT for uniformity (e.g. even with ```split_by_feature_attribute```, ```mean_sample```) etc we seems to have ```ACTION_DIMENSION```, it should probably be ```stack_by_sample|feature_attribute``` ?