TensorPlay
Reference guides
Copy
View MarkdownDownload .md

tensorplay.sparse

Tensor methods and sparse

Torch functions specific to sparse Tensors

tensorplay.functional.sparse_coo_tensor

tensorplay.sparse.sum

Sum of input's values over dim.

tensorplay.sparse.mm

Performs a matrix multiplication of a 2-D sparse COO/CSR tensor with a dense matrix.

tensorplay.sparse.spdiags

Constructs a sparse tensor from diagonals, mirroring torch.sparse.spdiags (and thereby scipy.sparse.spdiags).

Ask DeepWiki