- 输入1是一个两维tensor,[batch_size , tag_num]。
- 输入2是sentence len。
- 输出是一个二维的tensor , [batch_size , sentence_len]。
import tensorflow as tf
samples = tf.multinomial(tf.log([[10.,23.,34.],[34.,35.,89.]]), 5)
with tf.Session() as sess:
print(sess.run(samples))
>>[[1 2 1 2 1]
[0 1 2 2 0]]
网友评论