このページは Cloud Translation API によって翻訳されました。
Switch to English

tf.sequence_mask

TensorFlow 1つのバージョン GitHubの上のソースを表示

各セルの最初のN個の位置を示すマスクテンソルを返します。

場合lengths形状持つ[d_1, d_2, ..., d_n]得テンソルmask DTYPE有しdtypeと形状[d_1, d_2, ..., d_n, maxlen]とを、

 mask[i_1, i_2, ..., i_n, j] = (j < lengths[i_1, i_2, ..., i_n])
 

例:

 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]]]
 

lengths テンソル整数、その​​すべての値<= MAXLEN。
maxlen スカラ整数テンソル、返されたテンソルの最後の次元のサイズ。デフォルト値は最大値であるlengths
dtype 得られたテンソルの出力タイプ。
name OPの名前。

形状のマスクテンソルlengths.shape + (maxlen,)指定DTYPEにキャスト。

ValueError 場合maxlenスカラーではありません。