[Feature] Some improvements to VecNorm #10486

Job	Run time
build-wheel-windows (3.8, 3.8)	2m 9s
build-wheel-linux (3.8, cp38-cp38, --index-url https://download.pytorch.org/whl/cpu, "['cpu', '11...	1m 37s
build-wheel-windows (3.9, 3.9)	2m 42s
build-wheel-linux (3.9, cp39-cp39, --index-url https://download.pytorch.org/whl/cpu, "['cpu', '11...	1m 37s
build-wheel-windows (3.10, 3.10.3)	3m 2s
build-wheel-linux (3.10, cp310-cp310, --index-url https://download.pytorch.org/whl/cpu, "['cpu', ...	1m 43s
build-wheel-windows (3.11, 3.11)	2m 12s
build-wheel-linux (3.11, cp311-cp311, --index-url https://download.pytorch.org/whl/cpu, "['cpu', ...	1m 34s
test-wheel (linux, ubuntu-20.04, 3.8)	1m 47s
test-wheel (linux, ubuntu-20.04, 3.9)	1m 47s
test-wheel (linux, ubuntu-20.04, 3.10)	1m 45s
test-wheel (linux, ubuntu-20.04, 3.11)	1m 46s
test-wheel-windows (3.8)	11s
test-wheel-windows (3.9)	11s
test-wheel-windows (3.10)	11s
test-wheel-windows (3.11)	11s
	24m 25s

Provide feedback