テンソルフローのシーケンスマスク

Question

しばらくの間、私はテンソルフローの文書に従っています。 最近、私は実際には非常に便利な機能であるsequence_mask()を見つけました。 official documentationによると

、

sequence_mask(
    lengths, 
    maxlen=None, 
    dtype=tf.bool, 
    name=None 
) 

ここ

彼らはまた、2つの例を提供し、

tf.sequence_mask([1, 3, 2], 5) # [[True, False, False, False, False], 
           # [True, True, True, False, False], 
           # [True, True, False, False, False]] 

tf.sequence_mask([[1, 3],[2,0]]) # [[[True, False, False], 
            # [True, True, True]], 
            # [[True, True, False], 
            # [False, False, False]]] 

私は自分のコンピュータでそれらをテストしている間は、最初の例では、正常に実行されました。しかし、私が2番目の例を実行している間、エラーメッセージが表示されます。

ValueError: lengths must be 1D for sequence_mask 

だから何が問題なのですか？

Answer 1

あなたがインストールしたTesorflowバージョンのためだと思います。 Tensorflowバージョン1.4.0にアップデートして正常に動作するはずです。