org.tensorflow.op.math

Class SegmentMax<T extends Number>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.math.SegmentMax<T>

Type Parameters:

T - data type for output() output

All Implemented Interfaces:

Op, Operand<T>

@Operator(group="math")
public final class SegmentMax<T extends Number>
extends PrimitiveOp
implements Operand<T>

Computes the maximum along segments of a tensor.

Read [the section on segmentation](https://tensorflow.org/api_docs/python/tf/math#Segmentation) for an explanation of segments.

Computes a tensor such that \\(output_i = \max_j(data_j)\\) where `max` is over `j` such that `segment_ids[j] == i`.

If the max is empty for a given segment ID `i`, `output[i] = 0`.

For example:

 c = tf.constant([[1,2,3,4], [4, 3, 2, 1], [5,6,7,8]])
 tf.segment_max(c, tf.constant([0, 0, 1]))
 # ==> [[4, 3, 3, 4],
 #      [5, 6, 7, 8]]
 

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
Output<T>	asOutput() Returns the symbolic handle of a tensor.
static <T extends Number,U extends Number> SegmentMax<T>	create(Scope scope, Operand<T> data, Operand<U> segmentIds) Factory method to create a class wrapping a new SegmentMax operation.
Output<T>	output() Has same shape as data, except for dimension 0 which has size `k`, the number of segments.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static <T extends Number,U extends Number> SegmentMax<T> create(Scope scope,
                                                                       Operand<T> data,
                                                                       Operand<U> segmentIds)

Factory method to create a class wrapping a new SegmentMax operation.

Parameters:

scope - current scope

data -

segmentIds - A 1-D tensor whose size is equal to the size of `data`'s first dimension. Values should be sorted and can be repeated.

Returns:

a new instance of SegmentMax

output

public Output<T> output()

Has same shape as data, except for dimension 0 which has size `k`, the number of segments.

asOutput

public Output<T> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a tensor.

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.

Specified by:

asOutput in interface Operand<T extends Number>

See Also:

OperationBuilder.addInput(Output)