Searched refs:vec_to_tril_matrix (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributions/ |
| H A D | test_utils.py | 6 from torch.distributions.utils import tril_matrix_to_vec, vec_to_tril_matrix
25 tril_mat = vec_to_tril_matrix(vec, diag)
|
| H A D | test_transforms.py | 39 from torch.distributions.utils import tril_matrix_to_vec, vec_to_tril_matrix
427 lambda x: transform(vec_to_tril_matrix(x, diag=-1)),
|
| H A D | test_distributions.py | 106 vec_to_tril_matrix,
3934 x = vec_to_tril_matrix(x, -1)
|
| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | utils.py | 179 def vec_to_tril_matrix(vec: torch.Tensor, diag: int = 0) -> torch.Tensor: function
|
| H A D | transforms.py | 17 vec_to_tril_matrix,
835 r = vec_to_tril_matrix(x, diag=-1)
|