tf.segment_max函数:计算一个张量的最大沿段
2018-01-10 10:12 更新
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除外.