def to_tensor(rt, default_value=None, name=None, shape=None):
"""See `tf.RaggedTensor.to_tensor`."""
assert is_ragged(rt)
return Lambda(
lambda args: args[0].to_tensor(args[1], name=name, shape=shape))(
[rt, default_value])
def value_rowids(rt):
assert is_ragged(rt)
return Lambda(lambda rt: tf.identity(rt.value_rowids()))(rt)
def nested_row_splits(rt):
assert is_ragged(rt)
return Lambda(lambda rt: [tf.identity(s) for s in rt.nested_row_splits])(
rt)
def row_starts(rt):
assert is_ragged(rt)
return Lambda(lambda rt: tf.identity(rt.row_starts()))(rt)
def flat_values(rt):
assert is_ragged(rt)
return Lambda(lambda rt: tf.identity(rt.flat_values))(rt)
