torch.Tensor.to_sparse_coo

Tensor.to_sparse_coo()[源代码]

将张量转换为坐标格式。

示例:

>>> dense = torch.randn(5, 5)
>>> sparse = dense.to_sparse_coo()
>>> sparse._nnz()
25