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