tf.segment_prod 函数

segment_prod(
    data,
    segment_ids,
    name=None
)

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

沿张量的片段计算乘积.

计算的张量,例如： \(output_i = \prod_j data_j\),其中所述乘积是在j,例如：segment_ids[j] == i.

如果乘积对于给定ID i 的分段是空的,则应该有output[i] = 1.

函数参数

• 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_prod函数返回的是一个Tensor,它与data有相同的类型,与data具有相同的形状, 但大小为 k(线段的数目)的维度0除外.