tf.unsorted_segment_max

tf.unsorted_segment_max

tf.unsorted_segment_max

unsorted_segment_max(
    data,
    segment_ids,
    num_segments,
    name=None
)

Defined in tensorflow/python/ops/gen_math_ops.py.

Computes the Max along segments of a tensor.

Read the section on segmentation for an explanation of segments.

This operator is similar to the unsorted segment sum operator. Instead of computing the sum over segments, it computes the maximum such that:

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

If the maximum is empty for a given segment ID i, it outputs the smallest possible value for specific numeric type, output[i] = numeric_limits<T>::min().

Args:

• data: A Tensor. Must be one of the following types: float32, float64, int32, int64, uint8, int16, int8, uint16, half.
• segment_ids: A Tensor. Must be one of the following types: int32, int64. A 1-D tensor whose rank is equal to the rank of data's first dimension.
• num_segments: A Tensor of type int32.
• name: A name for the operation (optional).

Returns:

A Tensor. Has the same type as data. Has same shape as data, except for dimension 0 which has size num_segments.