tf.segment_max函數(shù):計算一個張量的最大沿段

2018-01-10 10:12 更新

tf.segment_max 函數(shù)

segment_max(
    data,
    segment_ids,
    name=None
)

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

計算一個張量的最大沿段.

計算一個張量:\(output_i = \ max_j(Data_j)\),其中max大于j,例如:segment_ids[j] == i.

如果給定段 ID 為 i 的最大值為空,則有output[i] = 0.

TensorFlow函數(shù)

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

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

函數(shù)返回值

tf.segment_max函數(shù)會返回Tensor,并且與data具有相同的類型. 但大小為k(線段的數(shù)目)的維度0除外.

以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號