torch.Tensor.crow_indices¶
- Tensor.crow_indices() IntTensor ¶
当
self
是布局为sparse_csr
的稀疏 CSR 张量时,返回包含self
张量压缩行索引的张量。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)