tf.ragged.row_splits_to_segment_ids
Создает сегментацию, соответствующую row_splits типа RaggedTensor.
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 | Отсортированный одномерный целочисленный тензор. splits[0] должен быть равен нулю. |
name | Префикс имени возвращаемого тензора (необязательный). |
out_type | Тип данных возвращаемого значения. По умолчанию — splits.dtype или tf.int64, если у splits не указан тип данных. |
| Возвращает | |
|---|---|
Отсортированный одномерный целочисленный тензор с shape=[splits[-1]] |
| Вызывает исключение | |
|---|---|
ValueError | Если splits имеет недопустимое значение. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/ragged/row_splits_to_segment_ids
