テンソルフロー::作戦::未分類セグメント合計

#include <math_ops.h>

テンソルのセグメントに沿って合計を計算します。

まとめ

セグメントの説明については、セグメンテーションに関するセクションを参照してください。

\(output[i] = {j...} data[j...]\) の合計がタプルj...を超えsegment_ids[j...] == iとなるようなテンソルを計算します。 SegmentSumとは異なり、 segment_ids並べ替える必要がなく、有効な値の全範囲内のすべての値をカバーする必要もありません。

指定されたセグメント ID iの合計が空の場合、 output[i] = 0 。指定されたセグメント ID iが負の場合、その値は削除され、セグメントの合計には追加されません。

num_segments個別のセグメント ID の数と等しくなければなりません。

c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]])
tf.unsorted_segment_sum(c, tf.constant([0, 1, 0]), num_segments=2)
# ==> [[ 5,  5, 5, 5],
#       [5,  6, 7, 8]]

引数:

  • スコープ:スコープオブジェクト
  • segment_ids: 形状がdata.shapeのプレフィックスであるテンソル。

戻り値:

  • Output : 最初のsegment_ids.rankディメンションを除き、データと同じ形状を持ちます。最初のsegment_ids.rankディメンションは、サイズがnum_segmentsの単一ディメンションに置き換えられます。

コンストラクターとデストラクター

UnsortedSegmentSum (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids, :: tensorflow::Input num_segments)

パブリック属性

operation
output

公共機能

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

パブリック属性

手術

Operation operation

出力

::tensorflow::Output output

公共機能

未分類セグメント合計

 UnsortedSegmentSum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input segment_ids,
  ::tensorflow::Input num_segments
)

ノード

::tensorflow::Node * node() const 

演算子::tensorflow::入力

 operator::tensorflow::Input() const 

演算子::tensorflow::出力

 operator::tensorflow::Output() const