Small change to LKJ to make the distribution more numpy-friendly (failure exposed by correlation_cholesky_test).
Small change to LKJ to make the distribution more numpy-friendly (failure exposed by correlation_cholesky_test).

PiperOrigin-RevId: 263589774

tensorflow/probability,tensorflow/probability