TensorFlow函數(shù)：tf.sparse_segment_sqrt_n

tf.sparse_segment_sqrt_n 函數(shù)

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

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

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

計(jì)算由 sqrt (N) 除以的張量的稀疏段的總和.
N 是要減少的線段的大小.

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

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

函數(shù)返回值：

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