torch.distributions.utils.tril_matrix_to_vec

torch.distributions.utils.tril_matrix_to_vec(mat, diag=0)

Convert a D x D matrix or a batch of matrices into a (batched) vector which comprises of lower triangular elements from the matrix in row order.

Return type:

Tensor