Skip to content

Sparse tensor use cases #10043

@weiyangfb

Description

@weiyangfb

We are working on to increase supports for sparse tensor. Currently we have summarized current state of sparse tensor and listed out sparse ops to support. We would like to collect sparse tensor use cases to facilitate the design decisions and prioritize TODO list according. It will be very helpful if you can post your use cases and desired sparse ops here or at the PyTorch Forum. Thanks!

I find these questions useful when writing use cases:

  • Where do I need sparse tensor? During training deep learning model?
  • Do I need autograd support for the sparse ops?

A possible example will be:

I am training model that has mul(Sparse, Dense) ops. I would like to have its forward and backward. I know there will be a dense gradient at the backward of mul, so here I am asking for a special kind of mul ops (called sparse_mul) that returns a sparse grad tensor and only keep the nnz's gradients.

Metadata

Metadata

Assignees

No one assigned

    Labels

    featureA request for a proper, new feature.module: sparseRelated to torch.sparsetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions