generate_pa_inputs_psm function

Generate probabilistic model inputs for partitioned survival model.