tensorflow 学习笔记-- tf.reduce_max、tf.sequence_mask
1、tf.reduce_max函數(shù)的作用:計算張量的各個維度上的元素的最大值。例子:
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)
結(jié)果為3
??? 2、tf.sequence_mask的作用是構(gòu)建序列長度的mask標(biāo)志 。 例子:
import tensorflow as tf
mask = tf.sequence_mask([1, 3, 2], 5)
with tf.Session() as sess:
mask = sess.run(mask)
print(mask)
結(jié)果是:
[[ True False False False False]
[ True True True False False]
[ True True False False False]]
??? 3、兩個函數(shù)結(jié)合使用:
? # 根據(jù)目標(biāo)序列長度,選出其中最大值,然后使用該值構(gòu)建序列長度的mask標(biāo)志,代碼:
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)
結(jié)果是:
[[ True False False]
[ True True True]
[ True True False]]
---------------------
作者:qq_28808697
來源:CSDN
原文:https://blog.csdn.net/qq_28808697/article/details/80648657
版權(quán)聲明:本文為博主原創(chuàng)文章,轉(zhuǎn)載請附上博文鏈接!
總結(jié)
以上是生活随笔為你收集整理的tensorflow 学习笔记-- tf.reduce_max、tf.sequence_mask的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: \r \r\n \t的区别
- 下一篇: PyTorch: 序列到序列模型(Seq