TensorFlow 构造文本mask

Dear 丶 2022-09-09 09:10 142阅读 0赞
  1. import tensorflow as tf
  2. input_sequence = tf.constant([[1,2,0,3],[1,0,2,0]],dtype=tf.float32)
  3. input_mask = tf.sign(input_sequence)
  4. sess = tf.Session()
  5. print(sess.run(input_mask))

print结果:

  1. [[1. 1. 0. 1.]
  2. [1. 0. 1. 0.]]

发表评论

表情:
评论列表 (有 0 条评论,142人围观)

还没有评论,来说两句吧...

相关阅读