TensorFlow函數(shù)：tf.unsorted_segment_max

tf.unsorted_segment_max函數(shù)

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

沿著張量的片段計(jì)算最大值.

此運(yùn)算符與未排序的段總和運(yùn)算符類似.它不是計(jì)算段的總和,而是計(jì)算最大值：

\(output_i = \ max_j data_j \),其中 max 超過了 j 這個值,segment_ids[j] == i.

如果給定分段 ID i 的最大值為空,則它會輸出特定數(shù)值類型的最小可能值,output[i] = numeric_limits<T>::min().

函數(shù)參數(shù)：

• 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：操作的名稱(可選).

函數(shù)返回值：

tf.unsorted_segment_max函數(shù)返回一個Tensor,它與 data 的類型相同.