tf.segment_sum详解

定义

tf.segment_sum(
    data,
    segment_ids,
    name=None
)

Computes the sum along segments of a tensor.
对输入数据data进行分割，并按对应下标进行求和。

输入数据

data为待分割数据
segment_ids为分割下标。segment_ids的大小应与data的第一个维度长度（k）相同。且segment_ids的值应小于k。
name是该操作命名。

详解

Computes a tensor such that (output_i = \sum_j data_j) where sum is over j such that segment_ids[j] == i.
If the sum is empty for a given segment ID i, output[i] = 0.
segment_ids中的值相当于对data数据第一维度的索引。输出数据的第一个维度（output[0]）为segment_ids==0的下标所对应的data的数据的求和。

示例

data = [5 1 7 2 3 4 1 3]
segment_ids = [0 2 4 1 5 2 4 1]
output = tf.segment_sum(
    data,
    segment_ids,
    name=None
)

• segment_ids的大小与data第一维度相同。（=8）且segment_ids的每个值<8。
• output[0] = data[segment_ids==0] = 5
• output[2] = data[segment_ids==2] = 1+4 = 5
• 对于segment_ids没有指定的下标（output[3]）直接等于0，output[3]=0

关注微信公众号“遥感头号”，获取更多有关python、深度学习、遥感的信息、学习笔记、资料