パブリック最終クラスSparseSegmentSqrtNGradV2
SparseSegmentSqrtN の勾配を計算します。
値が「indices」の一意のインデックスの数である次元 0 を除き、grad と同じ形状のテンソル「output」を返します。また、「indices」からの対応するインデックスを含むベクトル「sorted_unique_indices」も返します。
パブリックメソッド
static <T は数値を拡張、U は数値を拡張、V は数値を拡張> SparseSegmentSqrtNGradV2 <T, U> | |
出力<T> | 出力() |
出力<U> |
継承されたメソッド
パブリックメソッド
public static SparseSegmentSqrtNGradV2 <T, U> create (スコープスコープ、オペランド<T> grad、オペランド<U> インデックス、オペランド<V> セグメント ID、オペランド<Integer> DensityOutputDim0)
新しい SparseSegmentSqrtNGradV2 オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
卒業生 | 勾配は SparseSegmentSqrtN 演算に伝播されます。 |
インデックス | 対応する SparseSegmentSqrtN 演算に渡されるインデックス。 |
セグメントID | 対応する SparseSegmentSqrtN 演算に渡されるsegment_ids。 |
密出力Dim0 | SparseSegmentSqrtN 演算に渡される「データ」の次元 0。 |
戻り値
- SparseSegmentSqrtNGradV2 の新しいインスタンス