TensorFlow 学习(十二)—— 高级函数

来源:互联网 发布:美孚齿轮油632数据 编辑:程序博客网 时间:2024/05/18 00:04
  • tf.map_fn(fn, elems):接受一个函数对象,然后用该函数对象对集合(elems)中的每一个元素分别处理,

    def preprocessing_image(image, training):    image = ...    return imagedef preprocessing_images(images, training):    images = tf.map_fn(lambda image: preprocessing_image(image, training), images)    return images
0 0