阅读(6.3k) 书签 (0)

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.

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