tf.segment_max函数:计算一个张量的最大沿段

由 Carrie 创建, 最后一次修改 2018-01-10

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.

TensorFlow函数

函数参数

  • 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除外.

以上内容是否对您有帮助:
二维码
建议反馈
二维码