TensorFlow函数:tf.unsorted_segment_max
2018-04-13 10:03 更新
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 的类型相同.