Joins the elements of inputs
based on segment_ids
.
tf.raw_ops.UnsortedSegmentJoin(
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 = [['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']
Args | |
---|---|
inputs
|
A Tensor of type string . The input to be joined.
|
segment_ids
|
A Tensor . Must be one of the following types: int32 , int64 .
A tensor whose shape is a prefix of data.shape. Negative segment ids are not
supported.
|
num_segments
|
A Tensor . Must be one of the following types: int32 , int64 .
A scalar.
|
separator
|
An optional string . Defaults to "" .
The separator to use when joining.
|
name
|
A name for the operation (optional). |
Returns | |
---|---|
A Tensor of type string .
|