TensorFlow：tf.segment_mean函数

tf.segment_mean 函数

segment_mean(
    data,
    segment_ids,
    name=None
)

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

计算张量的沿段的平均值.

计算一个张量,例如：\(output_i = \ frac {\ sum_j data_j} {N} \),其中mean为j,segment_ids[j] == i并且N是求和的总值.

如果给定段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_mean函数会返回的是Tensor,并且与data有相同的类型,但大小为k(线段的数目)的维度0除外.