Update _augment.py #1011

Job	Run time
linux (linux.g5.4xlarge.nvidia.gpu, cuda, 11.8) / linux-job	54s
linux (linux.12xlarge, cpu) / linux-job	3m 46s
macos (macos-12) / macos-job	1m 58s
macos (macos-m1-12) / macos-job	1m 14s
windows (windows.4xlarge, cpu) / windows-job	2m 0s
windows (windows.g5.4xlarge.nvidia.gpu, cuda, 11.8) / windows-job	37s
	10m 29s

Provide feedback