Login Register

tf.contrib.rnn.AttentionCellWrapper

class tf.contrib.rnn.AttentionCellWrapper

Basic attention cell wrapper.

Implementation based on https://arxiv.org/pdf/1601.06733.pdf.

Links:

https://www.tensorflow.org/versions/master/api_docs/python/contrib.rnn.html#AttentionCellWrapper

doc_TensorFlow

2016-10-14 13:07:20

Comments

Leave a Comment

Please login to continue.

Popular Articles

tf.contrib.rnn.AttentionCellWrapper.__init__() tf.contrib.rnn.AttentionCellWrapper.__init__(cell, attn_length, attn_size=None, attn_vec_size=None, input_size=None, state_is_tuple=False) Create a

tf.contrib.rnn.AttentionCellWrapper.__call__() tf.contrib.rnn.AttentionCellWrapper.__call__(inputs, state, scope=None) Long short-term memory cell with attention (LSTMA).

tf.contrib.rnn.AttentionCellWrapper.output_size tf.contrib.rnn.AttentionCellWrapper.output_size

tf.contrib.rnn.AttentionCellWrapper.state_size tf.contrib.rnn.AttentionCellWrapper.state_size

tf.contrib.rnn.AttentionCellWrapper.zero_state() tf.contrib.rnn.AttentionCellWrapper.zero_state(batch_size, dtype) Return zero-filled state tensor(s). Args: batch_size: int, float, or unit Tensor