torch.distributions.utils.vec_to_tril_matrix

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

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

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

Tensor