View source on GitHub
|
Utility functions for ranking library.
Functions
de_noise(...): Returns a float Tensor as the de-noised counts.
gather_per_row(...): Gathers the values from input tensor based on per-row indices.
is_label_valid(...): Returns a boolean Tensor for label validity.
organize_valid_indices(...): Organizes indices in such a way that valid items appear first.
padded_nd_indices(...): Pads the invalid entries by valid ones and returns the nd_indices.
parse_keys_and_weights(...): Parses the encoded key to keys and weights.
ragged_to_dense(...): Converts given inputs from ragged tensors to dense tensors.
reshape_first_ndims(...): Reshapes the first n dims of the input tensor to new shape.
reshape_to_2d(...): Converts the given tensor to a 2-D Tensor.
shuffle_valid_indices(...): Returns a shuffle of indices with valid ones on top.
sort_by_scores(...): Sorts list of features according to per-example scores.
sorted_ranks(...): Returns an int Tensor as the ranks (1-based) after sorting scores.