![]() |
Returns a Tensor
of indices within each segment.
tft.segment_indices(
segment_ids, name=None
)
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.
|