テンソルのセグメントに沿って合計を計算します。
セグメントの説明については、[セグメンテーションに関するセクション](https://tensorflow.org/api_docs/python/tf/math#Segmentation) を参照してください。
\\(output_i = \sum_j data_j\\) のようなテンソルを計算します。ここで、合計は `j` を超え、`segment_ids[j] == i` になります。
指定されたセグメント ID `i` の合計が空の場合、`output[i] = 0` になります。
この操作は現在 jit_compile=True でのみサポートされていることに注意してください。
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <T, U は数値を拡張、V は数値を拡張> SegmentSumV2 <T> | |
出力<T> | 出力() 最初の `segment_ids.rank` 次元を除き、データと同じ形状を持ちます。最初の `segment_ids.rank` 次元は、サイズ `num_segments` を持つ単一の次元に置き換えられます。 |
継承されたメソッド
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static SegmentSumV2 <T> create (スコープスコープ、オペランド<T> データ、オペランド<U> セグメント ID、オペランド<V> numSegments)
新しい SegmentSumV2 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
セグメントID | サイズが「data」の最初の次元のサイズと等しい 1 次元テンソル。値は並べ替える必要があり、繰り返すことができます。値は「num_segments」未満である必要があります。 注意: 値は常に CPU 上で並べ替えられるように検証されますが、GPU 上では検証されません。 |
戻り値
- SegmentSumV2 の新しいインスタンス
public出力<T>出力()
最初の `segment_ids.rank` 次元を除き、データと同じ形状を持ちます。最初の `segment_ids.rank` 次元は、サイズ `num_segments` を持つ単一の次元に置き換えられます。