added mapping between subset idx and true idx in stochgradmom