SparseSegmentMeanGradV2

パブリック最終クラスSparseSegmentMeanGradV2

SparseSegmentMean の勾配を計算します。

値が「indices」の一意のインデックスの数である次元 0 を除き、grad と同じ形状のテンソル「output」を返します。また、「indices」からの対応するインデックスを含むベクトル「sorted_unique_indices」も返します。

パブリックメソッド

static <T extends Number, U extends Number, V extends Number> SparseSegmentMeanGradV2 <T, U>
create (スコープscope、オペランド<T> grad、オペランド<U>インデックス、オペランド<V>セグメントIds、オペランド<Integer>denseOutputDim0)
新しい SparseSegmentMeanGradV2 操作をラップするクラスを作成するファクトリ メソッド。
出力<T>
出力()
出力<U>
sortedUniqueIndices ()

継承されたメソッド

パブリックメソッド

public static SparseSegmentMeanGradV2 <T, U> create (スコープスコープ、オペランド<T> grad、オペランド<U> インデックス、オペランド<V> セグメント ID、オペランド<Integer> DensityOutputDim0)

新しい SparseSegmentMeanGradV2 操作をラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
卒業生勾配は SparseSegmentMean 演算に伝播されます。
インデックス対応する SparseSegmentMean 演算に渡されるインデックス。
セグメントID対応する SparseSegmentMean 演算に渡されるsegment_ids。
密出力Dim0 「データ」の次元 0 が SparseSegmentMean 演算に渡されました。
戻り値
  • SparseSegmentMeanGradV2 の新しいインスタンス

public出力<T>出力()

public出力<U> selectedUniqueIndices ()