d6dcffa15112f03d1bb08764740a78728e47e7d7,texar/models/tsf/ops.py,GreedyPointerDecoder,__init__,#GreedyPointerDecoder#Any#Any#Any#Any#Any#Any#Any#,190

Before Change

class GreedyPointerDecoder(PointerDecoder):
  def __init__(self, proj_layer, attn_layer, pointer_layer, enc_inputs,
               enc_outputs, embedding, output_keep_prob=0.5):
    super(PointerDecoder, self).__init_(proj_layer, attn_layer, pointer_layer,
                                        enc_inputs, enc_outputs, embedding)
  def next_inputs(self, t, p):
    word = tf.argmax(p, axis=1)
    next_inps = tf.nn.embeding_lookup(word, self._embedding)
    return next_inps, word

After Change

class GreedyPointerDecoder(PointerDecoder):
  def __init__(self, proj_layer, attn_layer, pointer_layer, enc_inputs,
               enc_outputs, embedding, output_keep_prob=0.5):
    super(GreedyPointerDecoder, self).__init__(
      proj_layer, attn_layer, pointer_layer, enc_inputs,
      enc_outputs, embedding)
  def next_inputs(self, t, p):
    word = tf.argmax(p, axis=1)
    next_inps = tf.nn.embedding_lookup(self._embedding, word)
    return next_inps, word

Instances