The matrix format after using ToSparseTensor()

[ZhouAo-ZA]

When we want to use SpMM mode to calculate GCN, data needs to be converted to sparse matrix format using ToSparcetensor. I want to know whether the matrix is CSR, CSC or COO format before executing SpMM？Similarly, before executing segment_csr ?

[rusty1s]

SparseTensor can represent any sparse format (COO, CSR, CSC). For SpMM, it will convert to CSR.

[ZhouAo-ZA]

thanks, got it!