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]]
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • TensorFlow Eager 教程 来源:madalinabuzau/tensorflow-eager-tut...
    布客飞龙阅读 11,697评论 1 17
  • TensorFlow 高效编程 原文:vahidk/EffectiveTensorflow 译者:FesianXu...
    布客飞龙阅读 6,057评论 0 45
  • 你在成长 你还是孩子 你会有自己是想法 有自己想做的事情 你不能因为你的一句喜欢 去限制另一个孩子的想法和自由 就...
    七八岁阅读 255评论 0 5
  • 我困在小屋里 四周都是冰冷的墙 伸手看见泥巴掉落 啪啪响成旋律 不发声地打节拍 手脚蜷缩在正中心 哪面墙都遥远
    7_涵澈阅读 307评论 0 0
  • 满目葱茏赤间黄, 忽看孤菊白如霜。 肤清骨洁真君子, 艳丽无争只沁香。
    刘红科阅读 846评论 10 50

友情链接更多精彩内容