def batch_sparse_tensor_dense_matmul(sp_a, b): '''Multiply sp_a by every row of b.''' return tf.map_fn(lambda b_i: tf.sparse_tensor_dense_matmul(sp_a, b_i), b)