TensorFlow函數(shù)：tf.sparse_segment_mean

tf.sparse_segment_mean 函數(shù)

sparse_segment_mean ( 
    data , 
    indices , 
    segment_ids , 
    name = None , 
    num_segments = None 
)

定義在：tensorflow/python/ops/math_ops.py.

請(qǐng)參閱指南：數(shù)學(xué)>分段

tf.sparse_segment_mean函數(shù)計(jì)算沿張量稀疏段的平均值.

與 SegmentMean 相同,但 segment_ids 可以有低于 data 第一維的秩,選擇維度 0 的子集,由 indices 指定.segment_ids 允許缺少 ID,在這種情況下,輸出將在這些索引中為零.在這些情況下,num_segments 用于確定輸出的大小.

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

• data：Tensor 數(shù)據(jù)將在輸出中組合.
• indices：一個(gè)一維的 Tensor,帶有索引 data,與 segment_ids 有相同的秩.
• segment_ids：一維的 Tensor,帶有索引輸入Tensor,其值應(yīng)該被排序并且可以重復(fù).
• name：操作的名稱(可選).
• num_segments：一個(gè)可選的 int32 標(biāo)量,指示輸出 Tensor 的大小.

函數(shù)返回值：

作為數(shù)據(jù)的形狀的一個(gè) tensor,除了大小為 k 的維度 0 之外,它通過(guò) num_segments 指定的段的數(shù)量或在 segments_ids 中推斷最后一個(gè)元素.