CYLphysics

@JustinS6626 , I also have same question about this, and I found in https://github.com/mit-han-lab/torchquantum/blob/main/torchquantum/measurement/measurements.py#L46 . `state_mag = qdev.get_states_1d().abs().detach().cpu().numpy()` This can extract the probability like qml.probs().