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