テンソルのセグメントに沿って合計を計算します。
セグメントの説明については、[セグメンテーションに関するセクション](https://tensorflow.org/api_docs/python/tf/math#Segmentation) を参照してください。
\\(output[i] = \sum_{j...} data[j...]\\) の合計が `segment_ids[j...] == i` となるタプル `j...` を超えるようなテンソルを計算します。 `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]]
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <T extends TType > UnsortedSegmentSum <T> | |
出力<T> | 出力() 最初の `segment_ids.rank` 次元を除き、データと同じ形状を持ちます。最初の `segment_ids.rank` 次元は、サイズ `num_segments` を持つ単一の次元に置き換えられます。 |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static UnsortedSegmentSum <T> create (スコープスコープ、オペランド<T> データ、オペランド<? extends TNumber > セグメント ID、オペランド<? extends TNumber > numSegments)
新しい UnsortedSegmentSum 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
セグメントID | 形状が `data.shape` の接頭辞であるテンソル。 |
戻り値
- UnsortedSegmentSum の新しいインスタンス
public出力<T>出力()
最初の `segment_ids.rank` 次元を除き、データと同じ形状を持ちます。最初の `segment_ids.rank` 次元は、サイズ `num_segments` を持つ単一の次元に置き換えられます。