torch.distributions.utils.tril_matrix_to_vec

torch.distributions.utils.tril_matrix_to_vec(mat, diag=0) [исходный код]

Преобразует матрицу D x D или пакет матриц в (пакетный) вектор, содержащий элементы нижнего треугольника матрицы в порядке строк.

Тип возвращаемого значения:

Tensor