b28653e110ad2bf780fb0cf083718f4a337a57bb,tensorflow_ranking/python/utils.py,,shuffle_valid_indices,#Any#Any#,71

Before Change

    [batch_size, list_size] tensor. The values in the last dimension are the
    indices for an element in the input tensor.
  
  is_valid = ops.convert_to_tensor(is_valid)
  is_valid.get_shape().assert_has_rank(2)
  output_shape = array_ops.shape(is_valid)
  rand = array_ops.where(is_valid,
                         random_ops.random_uniform(output_shape, seed=seed),
                         array_ops.ones(output_shape) * -1e-6)
  // shape(indices) = [batch_size, list_size]
  _, indices = nn_ops.top_k(rand, output_shape[1], sorted=True)
  // shape(batch_ids) = [batch_size, list_size]
  batch_ids = array_ops.ones_like(indices) * array_ops.expand_dims(

After Change

def shuffle_valid_indices(is_valid, seed=None):
  Returns a shuffle of indices with valid ones on top.
  return organize_valid_indices(is_valid, shuffle=True, seed=seed)


def organize_valid_indices(is_valid, shuffle=True, seed=None):
  Organizes indices in such a way that valid items appear first.

Instances