tfma.types.RaggedTensorValue

RaggedTensorValue encapsulates a batch of ragged tensor values.

values A np.ndarray of values.
nested_row_splits A list of np.ndarray values representing the row splits (one per dimension including the batch dimension).