美文网首页TensorFlow
Tensorflow: tf.reduce_max、tf.seq

Tensorflow: tf.reduce_max、tf.seq

作者: azim | 来源:发表于2018-07-31 10:20 被阅读385次

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

相关文章

网友评论

    本文标题:Tensorflow: tf.reduce_max、tf.seq

    本文链接:https://www.haomeiwen.com/subject/abecvftx.html