Chore: refactor PT compute_stat #6809

Job	Run time
Build source distribution	14s
determine-arm64-runner	1s
Build wheels for cp311-manylinux_x86_64	6m 30s
Build wheels for cp311-manylinux_x86_64	11m 0s
Build wheels for cp311-macosx_x86_64	5m 54s
Build wheels for cp311-macosx_arm64	3m 27s
Build wheels for cp311-win_amd64	6m 24s
Build wheels for cp310-manylinux_aarch64	5m 57s
build_docker (12)	3m 55s
build_docker (_cu11, 11)	3m 50s
build_pypi_index	7s
upload_pypi	0s
deploy_pypi_index	0s
Pass testing build wheels	1s
	47m 20s

Provide feedback