TensorFlow 2.0 RC is available Learn more

tft.segment_indices

tft.segment_indices(
    segment_ids,
    name=None
)

Returns a Tensor of indices within each segment.

segment_ids should be a sequence of non-decreasing non-negative integers that define a set of segments, e.g. [0, 0, 1, 2, 2, 2] defines 3 segments of length 2, 1 and 3. The return value is a Tensor containing the indices within each segment.

Example input: [0, 0, 1, 2, 2, 2] Example output: [0, 1, 0, 0, 1, 2]

Args:

  • segment_ids: A 1-d Tensor containing an non-decreasing sequence of non-negative integers with type tf.int32 or tf.int64.
  • name: (Optional) A name for this operation.

Returns:

A Tensor containing the indices within each segment.