tf.ragged.map_flat

TensorFlow 1 version

View source on GitHub

Generates the segmentation corresponding to a RaggedTensor row_splits.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.ragged.row_splits_to_segment_ids

tf.ragged.row_splits_to_segment_ids(
    splits, name=None, out_type=None
)

Returns an integer vector segment_ids, where segment_ids[i] == j if splits[j] <= i < splits[j+1]. Example:

print(tf.ragged.row_splits_to_segment_ids([0, 3, 3, 5, 6, 9]))
 tf.Tensor([0 0 0 2 2 3 4 4 4], shape=(9,), dtype=int64)

Args
`splits`	A sorted 1-D integer Tensor. `splits[0]` must be zero.
`name`	A name prefix for the returned tensor (optional).
`out_type`	The dtype for the return value. Defaults to `splits.dtype`, or `tf.int64` if `splits` does not have a dtype.

Returns
A sorted 1-D integer Tensor, with `shape=[splits[-1]]`

Raises
`ValueError`	If `splits` is invalid.

TensorFlow

tf

tf.audio

tf.autograph

tf.bitwise

tf.compat

tf.config

tf.data

tf.debugging

tf.distribute

tf.dtypes

tf.errors

tf.estimator

tf.experimental

tf.feature_column

tf.graph_util

tf.image

tf.initializers

tf.io

tf.keras

tf.linalg

tf.lite

tf.lookup

tf.losses

tf.math

tf.metrics

tf.nest

tf.nn

tf.optimizers

tf.quantization

tf.queue

tf.ragged

tf.random

tf.raw_ops

tf.saved_model

tf.sets

tf.signal

tf.sparse

tf.strings

tf.summary

tf.sysconfig

tf.test

tf.tpu

tf.train

tf.version

tf.xla

tf.ragged / map_flat_values

View aliases

Args

Returns

Raises