torch_mist.utils.indexing

Module Contents

Functions

_make_indexing_mask(→ torch.Tensor)
select_k_others(→ torch.Tensor)
_make_off_selection_indexing_mask(...)
matrix_off_diagonal(matrix, k)

torch_mist.utils.indexing._make_indexing_mask(n: int, k: int, device: torch.Tensor) → torch.Tensor

torch_mist.utils.indexing.select_k_others(x: torch.Tensor, k: int) → torch.Tensor

torch_mist.utils.indexing._make_off_selection_indexing_mask(n: int, k: int, device: torch.device) → Tuple[torch.Tensor, torch.Tensor]

torch_mist.utils.indexing.matrix_off_diagonal(matrix: torch.Tensor, k: int)