Minor style fixed to `tfp.math.pinv`. Defer expand_dims until needed; don't embed assertion if known statically.
Minor style fixed to `tfp.math.pinv`.
Defer expand_dims until needed; don't embed assertion if known statically.

PiperOrigin-RevId: 193073319

tensorflow/probability,tensorflow/probability