tf.sparse_matmul

Умножение матрицы "a" на матрицу "b".

tf.sparse_matmul(
    a, b, transpose_a=False, transpose_b=False, a_is_sparse=False,
    b_is_sparse=False, name=None
)

Вводные данные должны быть двумерными матрицами, а внутренняя размерность "a" должна совпадать с внешней размерностью "b". Обе "a" и "b" должны быть Tensors, а не SparseTensors. Данный оператор оптимизирован для случая, когда хотя бы одна из матриц "a" или "b" является разреженной, в том смысле, что у них большая доля нулевых значений. Критерий перехода от использования этого оператора к умножению плотных матриц на одной платформе составлял 30% нулевых значений в разреженной матрице.

Вычисление градиента данной операции будет использовать разреженность входного градиента только тогда, когда этот градиент происходит от Relu.