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> | |
Wyjście <T> | wyjście () |
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