check_sparse_tensor_invariants

class torch.sparse.check_sparse_tensor_invariants(enable=True) [source]

Инструмент для управления проверкой инвариантов разреженных тензоров.

Существуют следующие варианты для управления проверкой инвариантов разреженных тензоров при создании разреженных тензоров:

1. Использование контекстного менеджера:

   with torch.sparse.check_sparse_tensor_invariants():
       run_my_model()
   

2. Использование процедурного подхода:

   prev_checks_enabled = torch.sparse.check_sparse_tensor_invariants.is_enabled()
   torch.sparse.check_sparse_tensor_invariants.enable()
   
   run_my_model()
   
   if not prev_checks_enabled:
       torch.sparse.check_sparse_tensor_invariants.disable()
   

3. Использование декоратора функции:

   @torch.sparse.check_sparse_tensor_invariants()
   def run_my_model():
       ...
   
   run_my_model()
   

4. Использование параметра check_invariants в вызове конструктора разреженного тензора. Например:

   >>> torch.sparse_csr_tensor([0, 1, 3], [0, 1], [1, 2], check_invariants=True)
   Traceback (most recent call last):
     File "<stdin>", line 1, in <module>
   RuntimeError: `crow_indices[..., -1] == nnz` is not satisfied.
   

static disable() [source]

Отключение проверки инвариантов разреженных тензоров в конструкторах разреженных тензоров.

См. torch.sparse.check_sparse_tensor_invariants.enable() для получения дополнительной информации.

static enable() [source]

Включение проверки инвариантов разреженных тензоров в конструкторах разреженных тензоров.

Примечание

По умолчанию проверки инвариантов разреженных тензоров отключены. Используйте torch.sparse.check_sparse_tensor_invariants.is_enabled() для получения текущего состояния проверки инвариантов разреженных тензоров.

Примечание

Флаг проверки инвариантов разреженных тензоров действует для всех конструкторов разреженных тензоров, как в Python, так и в ATen.

Флаг может быть локально переопределен параметром check_invariants в функциях-конструкторах разреженных тензоров.

static is_enabled() [source]

Возвращает True, если проверка инвариантов разреженных тензоров включена.

Примечание

Используйте torch.sparse.check_sparse_tensor_invariants.enable() или torch.sparse.check_sparse_tensor_invariants.disable() для управления состоянием проверок инвариантов разреженных тензоров.