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(), чтобы управлять состоянием проверки инвариантов разреженных тензоров.