tf.ragged.row_splits_to_segment_ids
Генерирует сегментацию, соответствующую RaggedTensor row_splits.
tf.ragged.row_splits_to_segment_ids(
splits, name=None, out_type=None
)
Возвращает целочисленный вектор segment_ids, где segment_ids[i] == j если splits[j] <= i < splits[j+1]. Пример:
ragged.row_splits_to_segment_ids([0, 3, 3, 5, 6, 9]).eval()
[ 0 0 0 2 2 3 4 4 4 ]
| Аргументы |
|---|
splits | Отсортированный целочисленный тензор 1-D. splits[0] должен быть равен нулю. |
name | Префикс имени для возвращаемого тензора (необязательно). |
out_type | Тип данных для возвращаемого значения. По умолчанию splits.dtype, или tf.int64, если у splits нет типа данных. |
| Возвращает |
|---|
Отсортированный целочисленный тензор 1-D, с shape=[splits[-1]] |
| Возможные исключения |
|---|
ValueError | Если splits некорректен. |
