working on n-d broadcasting with axis=None