FindDefinition / cumm

CUda Matrix Multiply library.
Apache License 2.0
70 stars 9 forks source link

I know that there are functions for tensor2tv in the cumm library, so how do I re-transform tv back into a tensor and keep its gradient? #24

Open kenfishball opened 11 months ago

kenfishball commented 11 months ago

I know that there are functions for tensor2tv in the cumm library, so how do I re-transform tv back into a tensor and keep its gradient?