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]. Пример:
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)
| Аргументы |
|---|
splits | Отсортированный целочисленный тензор 1-D. splits[0] должен быть равен нулю. |
name | Префикс имени для возвращаемого тензора (необязательно). |
out_type | Тип данных для возвращаемого значения. По умолчанию splits.dtype, или tf.int64, если у splits нет типа данных. |
| Возвращаемое значение |
|---|
Отсортированный целочисленный тензор 1-D, с shape=[splits[-1]] |
| Исключения |
|---|
ValueError | Если splits некорректен. |
