Update beta sampling code in augment.py#13525

copybara-service[bot] merged 2 commits intomastertensorflow/models:masterfrom

LakshmiKalaKadali-patch-4tensorflow/models:LakshmiKalaKadali-patch-4Copy head branch name to clipboard