tf.unsorted_segment_max

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

Defined in generated file: tensorflow/python/ops/gen_math_ops.py.

Computes the maximum 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 found (here). 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 the specific numeric type, output[i] = numeric_limits<T>::lowest().

Args:

  • data: A Tensor. Must be one of the following types: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64.
  • 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. Must be one of the following types: int32, int64.
  • name: A name for the operation (optional).

Returns:

A Tensor. Has the same type as data.