dense tensor 互转sparse tensor

dense转sparse

def arr2sparse(arr):
    arr_tensor = tf.constant(np.array(arr))
    arr_idx = tf.where(tf.not_equal(arr_tensor, 0))
    arr_sparse = tf.SparseTensor(arr_idx, tf.gather_nd(arr_tensor, arr_idx), arr_tensor.get_shape())
    return arr_sparse

sparse转dense

sparse_to_dense(sparse_indices, output_shape, sparse_values)