torch.sparse

若API未标明支持情况,则代表该API的支持情况待验证。

API名称

是否支持

限制与说明

Tensor.to_sparse_coo

否

  

Tensor.coalesce

     

Tensor.sparse_resize_

     

Tensor.sparse_resize_and_clear_

     

Tensor.is_coalesced

     

Tensor.crow_indices

否

  

Tensor.col_indices

否

  

Tensor.row_indices

否

  

Tensor.ccol_indices

否

  

torch.sparse_compressed_tensor

否

  

torch.sparse.sum

     

torch.sparse.addmm

     

torch.sparse.sampled_addmm

     

torch.sparse.mm

否

  

torch.sspaddmm

     

torch.hspmm

     

torch.smm

     

torch.sparse.softmax

     

torch.sparse.log_softmax

     

torch.sparse.spdiags

否

  

torch.sparse.check_sparse_tensor_invariants

否