Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision Upstream anchor labels

tensorplay.sparse.mm tensorplay.sparse.spdiags tensorplay.sparse.sum

API symbolssparse

Copy

View Markdown Download .md

tensorplay.sparse.mm

tensorplay.sparse.mm(sparse, dense)[source]: Performs a matrix multiplication of a 2-D sparse COO/CSR tensor with a dense matrix. Equivalent to torch.sparse.mm.

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.softshrink

Previous Page

tensorplay.sparse.spdiags

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