TensorFlow函数：tf.unsorted_segment_max

tf.unsorted_segment_max函数

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

沿着张量的片段计算最大值.

此运算符与未排序的段总和运算符类似.它不是计算段的总和,而是计算最大值：

\(output_i = \ max_j data_j \),其中 max 超过了 j 这个值,segment_ids[j] == i.

如果给定分段 ID i 的最大值为空,则它会输出特定数值类型的最小可能值,output[i] = numeric_limits<T>::min().

函数参数：

• data：一个 Tensor,必须是下列类型之一：float32,float64,int32,uint8,int16,int8,int64,bfloat16,uint16,half,uint32,uint64.
• segment_ids：一个 Tensor,必须是以下类型之一：int32,int64；一维张量,其秩等于 data 第一维的秩.
• num_segments：一个 Tensor,必须是以下类型之一：int32,int64.
• name：操作的名称(可选).

函数返回值：

tf.unsorted_segment_max函数返回一个Tensor,它与 data 的类型相同.