tf.strings.unsorted_segment_join  |  TensorFlow v2.16.1 (original) (raw)

tf.strings.unsorted_segment_join

Stay organized with collections Save and categorize content based on your preferences.

Joins the elements of inputs based on segment_ids.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.strings.unsorted_segment_join

tf.strings.unsorted_segment_join(
    inputs, segment_ids, num_segments, separator='', name=None
)

Computes the string join along segments of a tensor.

Given segment_ids with rank N and data with rank N+M:

output[i, k1...kM] = strings.join([data[j1...jN, k1...kM])

where the join is over all [j1...jN] such that segment_ids[j1...jN] = i.

Strings are joined in row-major order.

For example:

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)