torch.Tensor.col_indices

Tensor.col_indices() → IntTensor

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

Пример::

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