I have two sparse tensors
b of sizes
(A, B, C, D) and
(A, B) respectively.
How do I multiply the two of them?
I don’t want to copy
b across a new third and fourth dimension
D times because that wastes memory and also takes compute time. I also don’t want to convert both to dense and back to sparse (although converting
b to dense at some point would be fine; it’s just
a that takes time to convert).