PaddingFIFOQueueから動的シーケンスの長さを取得

Question

私はtf.PaddingFIFOQueueまたはtf.contrib.data.PaddedBatchDatasetを使用してさまざまな長さのシーケンスを送り、dequeue_manyを使用して、0パッドのバッチを取り出します。

このバッチのシーケンス長を取得する一般的な方法はありますか？

私の現在の解決策は、系列長をキューに追加入力として明示的に提供することです。すなわち、私はtf.PaddingFIFOQueue(names=["data", "seq_length"], ...)のようなsthを持っています。私もtf.ones_like()を使用することができますが、私の現在の方法は安く簡単です。しかし、それが正規/標準的な方法であるのか、それとも別の方法があるのだろうかと思います。

Answer 1

dataとseq_lengthをタプル（またはリスト）に結合し、そのタプルをキューにプッシュすることができます。

import tensorflow as tf 
sess = tf.InteractiveSession() 
q = tf.PaddingFIFOQueue(capacity=10, dtypes=[tf.int32, tf.int32], shapes=[[], [None]]) 
eq1 = q.enqueue([1, [1]]) 
eq2 = q.enqueue([2, [2,3]]) 
eq3 = q.enqueue([3, [4,5,6]]) 
dq = q.dequeue() 
sess.run(eq1) 
sess.run(eq2) 
sess.run(eq3) 
sess.run(dq) # [1, array([1], dtype=int32)] 
sess.run(dq) # [2, array([2, 3], dtype=int32)] 
sess.run(dq) # [3, array([4, 5, 6], dtype=int32)]