TensorFlow：tf.segment_prod函數(shù)

tf.segment_prod 函數(shù)

segment_prod(
    data,
    segment_ids,
    name=None
)

請參閱指南：數(shù)學(xué)函數(shù)>分割

沿張量的片段計(jì)算乘積.

計(jì)算的張量,例如： \(output_i = \prod_j data_j\),其中所述乘積是在j,例如：segment_ids[j] == i.

如果乘積對于給定ID i 的分段是空的,則應(yīng)該有output[i] = 1.

函數(shù)參數(shù)

• data：一個(gè)Tensor；必須是下列類型之一：float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.
• segment_ids：一個(gè)Tensor；必須是以下類型之一：int32,int64；一維張量,其秩等于data第一維的秩.值應(yīng)該被排序,并且可以是重復(fù)的.
• name：操作的名稱(可選).

函數(shù)返回值

tf.segment_prod函數(shù)返回的是一個(gè)Tensor,它與data有相同的類型,與data具有相同的形狀, 但大小為 k(線段的數(shù)目)的維度0除外.