2784339dddeecd94dbab074140ba8e478b42f653,opennmt/decoders/self_attention_decoder.py,SelfAttentionDecoder,_run,#SelfAttentionDecoder#Any#Any#Any#Any#Any#Any#Any#,87

Before Change

        memory_sequence_length = (memory_sequence_length,)
      memory_mask = []
      for mem, mem_length in zip(memory, memory_sequence_length):
        mem_mask = tf.sequence_mask(mem_length, maxlen=tf.shape(mem)[1], dtype=tf.float32)
        mem_mask = tf.expand_dims(mem_mask, 1)
        memory_mask.append(mem_mask)

    // Run each layer.
    new_cache = []

After Change

      if not isinstance(memory_sequence_length, (list, tuple)):
        memory_sequence_length = (memory_sequence_length,)
      memory_mask = [
          tf.sequence_mask(mem_length, maxlen=tf.shape(mem)[1])
          for mem, mem_length in zip(memory, memory_sequence_length)]

    // Run each layer.
    new_cache = []

Instances