Моя модель NN содержит операцию sparse.mm (), и когда я пытаюсь тренироваться, я получаю следующую ошибку: RuntimeError: Expected object of backend CPU but got backend SparseCPU for argument #2 'mat2'
Я видел различные сообщения, связанные с этим, но мне неясно, какчто обходной путь / решение отличается от преобразования его в плотную матрицу и выполнения стандартных плотных операций.Идея состоит в том, чтобы использовать torch.sparse.mm () для его производительности.
Спасибо!