Use torch.allclose to check tensor equality (#58429)
Summary:
This fixes test_lkj_cholesky_log_prob if default codepath is used
I.e. test is executed as follows:
```
ATEN_CPU_CAPABILITY=default python3 distributions/test_distributions.py -v -k test_lkj_cholesky_log_prob
```
Fixes https://github.com/pytorch/pytorch/issues/58381
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58429
Reviewed By: neerajprad
Differential Revision: D28484340
Pulled By: malfet
fbshipit-source-id: 32afcc75e5250f5a11d66b4fa194ea1c784454a6
