TensorFlow:tf.segment_mean函数

2018-01-11 10:12 更新

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.

TensorFlow函数

函数参数

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


以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号