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 = [['Y', 'q', 'c'], ['Y', '6', '6'], ['p', 'G', 'a']]
output_array = string_ops.unsorted_segment_join(inputs=inputs,
                                                segment_ids=[1, 0, 1],
                                                num_segments=2,
                                                separator=':'))
# output_array ==> [['Y', '6', '6'], ['Y:p', 'q:G', 'c:a']]


inputs = ['this', 'is', 'a', 'test']
output_array = string_ops.unsorted_segment_join(inputs=inputs,
                                                segment_ids=[0, 0, 0, 0],
                                                num_segments=1,
                                                separator=':'))
# output_array ==> ['this:is:a:test']