tensorflow dynamic_rnn() decay_steps,decay_rate,embedding_lookup( )

来源:互联网 发布:suse centos 区别 编辑:程序博客网 时间:2024/06/05 08:40
  • dynamic_rnn()为了设置不同batch不同大小
  • decay_steps,decay_rate是为了momentum
  • embedding_lookup( )是为了embedding
embedding = tf.Variable(np.identity(5, dtype=np.int32))input_embedding = tf.nn.embedding_lookup(embedding, input_ids)
原创粉丝点击