search

cai4cai / torchsparsegradutils

A collection of utility functions to work with PyTorch sparse tensors
Apache License 2.0
24 stars 3 forks source link

Added simple workarounds for gather_mm and segment_mm #57

Closed tvercaut closed 1 month ago

tvercaut commented 1 month ago

See #56