tf.segment_max 函数

segment_max(
    data,
    segment_ids,
    name=None
)

请参阅指南：数学函数>分割

计算一个张量的最大沿段.

计算一个张量：\(output_i = \ max_j(Data_j)\),其中max大于j,例如：segment_ids[j] == i.

如果给定段 ID 为 i 的最大值为空,则有output[i] = 0.

函数参数

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

函数返回值

tf.segment_max函数会返回Tensor,并且与data具有相同的类型. 但大小为k(线段的数目)的维度0除外.