torch.Tensor.crow_indices

Tensor.crow_indices() → IntTensor

Возвращает тензор, содержащий сжатые индексы строк тензора self, когда self является разреженным тензором CSR с раскладкой sparse_csr. Тензор crow_indices имеет размерность строго (self.size(0) + 1) и тип int32 или int64. При использовании процедур MKL, таких как умножение разреженных матриц, необходимо использовать индексацию int32 для избежания приведения типов и потенциальной потери информации.

Пример::

>>> csr = torch.eye(5,5).to_sparse_csr()
>>> csr.crow_indices()
tensor([0, 1, 2, 3, 4, 5], dtype=torch.int32)