1. 程式人生 > >tensorflow 學習筆記-- tf.reduce_max、tf.sequence_mask

tensorflow 學習筆記-- tf.reduce_max、tf.sequence_mask

    1、tf.reduce_max函式的作用:計算張量的各個維度上的元素的最大值。例子:

import tensorflow as tf
max_value = tf.reduce_max([1, 3, 2])
with tf.Session() as sess:
    max_value = sess.run(max_value)
    print(max_value)
結果為3

    2、tf.sequence_mask的作用是構建序列長度的mask標誌 。 例子:

import tensorflow as tf
mask = tf.sequence_mask([1, 3, 2], 5)
with tf.Session() as sess:
    mask = sess.run(mask)
    print(mask)

結果是:

[[ True False False False False]
 [ True  True  True False False]
 [ True  True False False False]]

    3、兩個函式結合使用:

  # 根據目標序列長度,選出其中最大值,然後使用該值構建序列長度的mask標誌,程式碼:

import tensorflow as tf
max_value = tf.reduce_max([1, 3, 2])
mask = tf.sequence_mask([1, 3, 2], max_value)
with tf.Session() as sess:
    mask = sess.run(mask)
    print(mask)

結果是:

[[ True False False]
 [ True  True  True]
 [ True  True False]]