torch.distributions.utils.vec_to_tril_matrix

torch.distributions.utils.vec_to_tril_matrix(vec, diag=0)

Convert a vector or a batch of vectors into a batched D x D lower triangular matrix containing elements from the vector in row order.

Return type:

Tensor