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)