More useful error messages in LKJ.
More useful error messages in LKJ.

The default number of elements that `assert_less` and friends report
is 3, which often leads to messages like "Expected everything in x to
be <= 1.  First elements of x: 0.8, 0.7, 0.9."

Printing more has a higher chance of getting at least one of the
elements that actually violates the assertion.

PiperOrigin-RevId: 301439981

tensorflow/probability,tensorflow/probability