Revert "[BC breaking] Remove assert_allclose (#164560)"

This reverts commit eac53545ce48b33d5aad93e8b027f93160a50c05.

Reverted https://github.com/pytorch/pytorch/pull/164560 on behalf of https://github.com/pytorch-auto-revert due to Reverted automatically by pytorch's autorevert, to avoid this behaviour add the tag autorevert: disable ([comment](https://github.com/pytorch/pytorch/pull/164560#issuecomment-5393238614))