Learning-based method for k-space trajectory design in MRI