SparseSegmentSqrtNGradV2

publiczna klasa końcowa SparseSegmentSqrtNGradV2

Oblicza gradienty dla SparseSegmentSqrtN.

Zwraca tensor „wyjście” o takim samym kształcie jak grad, z wyjątkiem wymiaru 0, którego wartość jest liczbą unikalnych indeksów w „indeksach”. Zwraca również wektor „sorted_unique_indices” zawierający odpowiednie indeksy z „indices”.

Metody publiczne

statyczny <T rozszerza liczbę, U rozszerza liczbę, V rozszerza liczbę> SparseSegmentSqrtNGradV2 <T, U>
utwórz (zakres zakresu , argument <T> grad, indeksy argumentu <U>, argument <V> segmentIds, argument <Integer> gęstyOutputDim0)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SparseSegmentSqrtNGradV2.
Wyjście <T>
Wyjście <U>

Metody dziedziczone

Metody publiczne

public static SparseSegmentSqrtNGradV2 <T, U> create ( Zakres zakresu, Operand <T> grad, Operand <U> indeksy, Operand <V> segmentIds, Operand <Integer> gęstyOutputDim0)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SparseSegmentSqrtNGradV2.

Parametry
zakres aktualny zakres
absolwent gradient propagowany do operacji SparseSegmentSqrtN.
indeksy indeksy przekazywane do odpowiedniego SparseSegmentSqrtN op.
identyfikatory segmentu segment_ids przekazane do odpowiedniej operacji SparseSegmentSqrtN.
gęstyWyjścieDim0 wymiar 0 „danych” przekazanych do SparseSegmentSqrtN op.
Zwroty
  • nowa instancja SparseSegmentSqrtNGradV2

publiczne wyjście <T> wyjście ()

publiczne dane wyjściowe <U> sortedUniqueIndices ()