若API未标明支持情况则代表该API暂无验证结果,待验证后更新表格。
API名称 |
是否支持 |
限制与说明 |
---|---|---|
Tensor.is_sparse_csr |
||
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 |
||
torch.sparse.check_sparse_tensor_invariants.disable |
||
torch.sparse.check_sparse_tensor_invariants.enable |
||
torch.sparse.check_sparse_tensor_invariants.is_enabled |
||
torch.sparse.as_sparse_gradcheck |