TensorFlow:tf.segment_prod函数
2018-01-12 10:09 更新
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除外.