Add Initial momentum information to HMC kernel reuslt and better naming of the last momentum.
Add Initial momentum information to HMC kernel reuslt and better naming of the last momentum.

PiperOrigin-RevId: 311862567

tensorflow/probability,tensorflow/probability