tensorflow学习笔记(四十四):用Queue保存更复杂的关系

当查看 tensorflow 中提供的 Queue 的类的时候,会发现有个 dtypes 参数,这个参数代表 Queue 中元素是什么类型的,如果 dtypes=[tf.string, tf.int64] ,这个表示,Queue 中每个元素是 (string, int)

import tensorflow as tf

queue = tf.FIFOQueue(capacity=100, dtypes=[tf.string, tf.int64])

# enqueue_many 的写法,两个元素放在两个列表里。
en_m = queue.enqueue_many([['hello', 'world'], [1, 2]])

# enqueue 的写法
en = queue.enqueue(['hello', 1])
deq = queue.dequeue()

with tf.Session() as sess:
    sess.run(en_m)
    print(sess.run(deq))

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区