-
Notifications
You must be signed in to change notification settings - Fork 27.1k
Description
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.