@Operator public final class UnsortedSegmentJoin extends PrimitiveOp implements Operand<String>
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']
Modifier and Type | Class and Description |
---|---|
static class |
UnsortedSegmentJoin.Options
Optional attributes for
UnsortedSegmentJoin |
operation
Modifier and Type | Method and Description |
---|---|
Output<String> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T extends Number,U extends Number> |
create(Scope scope,
Operand<String> inputs,
Operand<T> segmentIds,
Operand<U> numSegments,
UnsortedSegmentJoin.Options... options)
Factory method to create a class wrapping a new UnsortedSegmentJoin operation.
|
Output<String> |
output() |
static UnsortedSegmentJoin.Options |
separator(String separator) |
equals, hashCode, op, toString
public static <T extends Number,U extends Number> UnsortedSegmentJoin create(Scope scope, Operand<String> inputs, Operand<T> segmentIds, Operand<U> numSegments, UnsortedSegmentJoin.Options... options)
scope
- current scopeinputs
- The input to be joined.segmentIds
- A tensor whose shape is a prefix of data.shape. Negative segment ids are not
supported.numSegments
- A scalar.options
- carries optional attributes valuespublic static UnsortedSegmentJoin.Options separator(String separator)
separator
- The separator to use when joining.public Output<String> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<String>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.