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)
微信
公众号
扫码关注公众号