tfr.utils.ragged_to_dense

Converts given inputs from ragged tensors to dense tensors.

tfr.utils.ragged_to_dense(
    labels, predictions, weights
)

Args
labels	A tf.RaggedTensor of the same shape as predictions representing relevance.
predictions	A tf.RaggedTensor with shape [batch_size, (list_size)]. Each value is the ranking score of the corresponding example.
weights	An optional tf.RaggedTensor of the same shape of predictions or a tf.Tensor of shape [batch_size, 1]. The former case is per-example and the latter case is per-list.

Returns
A tuple (labels, predictions, weights, mask) of dense tf.Tensors.