Update beta sampling code in augment.py#13491

dibyaghosh wants to merge 1 commit intotensorflow:mastertensorflow/models:masterfrom

dibyaghosh:masterdibyaghosh/models:masterCopy head branch name to clipboard