Converts (batch of) vectors to (batch of) lower-triangular scale matrices.
tfp.trainable_distributions.tril_with_diag_softplus_and_shift( x, diag_shift=1e-05, name=None )
x: (Batch of)
Tensorrepresenting vectors which will be transformed into lower-triangular scale matrices with positive diagonal elements. Rightmost shape
nmust be such that
n = dims * (dims + 1) / 2for some positive, integer
softplustransformation of diagonal elements. Default value:
name_scopename for operations created by this function. Default value:
scale_tril: (Batch of) lower-triangular
x.dtypeand rightmost shape
n = dims * (dims + 1) / 2where
n = x.shape[-1].