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)