27185a3ca428d1298811a6c048870d992c05a0d4,opennmt/utils/decoding.py,,_gather_from_word_indices,#Any#Any#,544

Before Change

def _gather_from_word_indices(tensor, indices):
  Index the depth dim of a 2D tensor.
  output_shape = misc.shape_list(indices)
  batch_size = tf.shape(tensor)[0]
  num_indices = tf.size(indices) // batch_size
  batch_pos = tf.range(batch_size * num_indices) // num_indices
  tensor = tf.gather_nd(tensor, tf.stack([batch_pos, tf.reshape(indices, [-1])], axis=-1))
  tensor = tf.reshape(tensor, output_shape)
  return tensor

def _lengths_from_ids(ids, end_id):
  Compute sequence lengths from word ids.
  lengths = tf.not_equal(ids, end_id)

After Change

def _gather_from_word_indices(tensor, indices):
  Index the depth dim of a 2D tensor.
  return tf.gather(tensor, indices, axis=-1, batch_dims=1)

def _lengths_from_ids(ids, end_id):
  Compute sequence lengths from word ids.
  lengths = tf.not_equal(ids, end_id)

Instances