Can you please add a `Notes` section that mentions that the subset can only happen over a single axis?