TensorFlow数学函数:tf.segment_sum函数

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

tf.segment_sum 函数

segment_sum(
    data,
    segment_ids,
    name=None
)

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

沿张量的片段计算总和.

计算的张量,例如 \(output_i = \sum_j data_j\),其中求和是在j上,例如:segment_ids[j] == i.

如果给定ID i 的分段的和为空,则有output[i] = 0.

TensorFlow函数

函数参数

  • data:一个Tensor;必须是下列类型之一:float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.
  • segment_ids:一个Tensor;必须是以下类型之一:int32,int64;一维张量,其秩等于data第一维的秩;值应该被排序,并且可以是重复的.
  • name:操作的名称(可选).

函数返回值

tf.segment_sum函数返回的是一个Tensor,它与data有相同的类型,与data具有相同的形状, 但大小为 k(段的数目)的维度0除外.

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