Есть ли способ эффективно выполнять действия с разреженными матрицами, когда у меня сложные операции? (на самом деле я даже не уверен, как мой пример может быть упрощен до чего-либо без нарезки)
В этом примере я пытаюсь создать несколько входов, входящих в пакет (b) со сложным вектором ядра, который отличается для каждого примера в пакете (так как он был построен весами, умноженными на другой вход). Входы плотные, ядро редкое.
Пример:
outputs = special_math_ops.einsum("bi,bnilo->bnlo", x, kernel)