TensorFlow：tf.segment_mean函數(shù)

tf.segment_mean 函數(shù)

segment_mean(
    data,
    segment_ids,
    name=None
)

請(qǐng)參閱指南：數(shù)學(xué)函數(shù)>分割

計(jì)算張量的沿段的平均值.

計(jì)算一個(gè)張量,例如：\(output_i = \ frac {\ sum_j data_j} {N} \),其中mean為j,segment_ids[j] == i并且N是求和的總值.

如果給定段ID i的平均值為空,則有output[i] = 0.

函數(shù)參數(shù)

• data：是一個(gè)Tensor；必須是下列類型之一：float32,float64,int32,int64,uint8,int16,int8,uint16,half.
• segment_ids：一個(gè)Tensor；必須是以下類型之一：int32,int64；一維張量,其秩等于data第一維的秩；值應(yīng)該被排序并且可以重復(fù).
• name：操作的名稱(可選).

函數(shù)返回值

該tf.segment_mean函數(shù)會(huì)返回的是Tensor,并且與data有相同的類型,但大小為k(線段的數(shù)目)的維度0除外.