tf.strings.unsorted_segment_join
Объединяет элементы inputs по segment_ids.
tf.strings.unsorted_segment_join(
inputs, segment_ids, num_segments, separator='', name=None
)
Вычисляет объединение строк по сегментам тензора.
Пусть segment_ids с рангом N и data с рангом N+M:
output[i, k1...kM] = strings.join([data[j1...jN, k1...kM])
где объединение выполняется по всем [j1...jN], таким что segment_ids[j1...jN] = i.
Строки объединяются в порядке следования строк в матрице.
Например:
inputs = ['this', 'a', 'test', 'is']
segment_ids = [0, 1, 1, 0]
num_segments = 2
separator = ' '
tf.strings.unsorted_segment_join(inputs, segment_ids, num_segments,
separator).numpy()
array([b'this is', b'a test'], dtype=object)inputs = [['Y', 'q', 'c'], ['Y', '6', '6'], ['p', 'G', 'a']]
segment_ids = [1, 0, 1]
num_segments = 2
tf.strings.unsorted_segment_join(inputs, segment_ids, num_segments,
separator=':').numpy()
array([[b'Y', b'6', b'6'],
[b'Y:p', b'q:G', b'c:a']], dtype=object)| Аргументы | |
|---|---|
inputs | Список объектов tf.Tensor типа tf.string. |
segment_ids | Тензор, форма которого является префиксом inputs.shape, и тип которого должен быть tf.int32 или tf.int64. Отрицательные идентификаторы сегментов не поддерживаются. |
num_segments | Скаляр типа tf.int32 или tf.int64. Должен быть неотрицательным и больше любого идентификатора сегмента. |
separator | Разделитель для объединения. По умолчанию "". |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Тензор tf.string, представляющий конкатенированные значения с указанным разделителем. |
© 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/strings/unsorted_segment_join
