publiczna klasa końcowa SparseSegmentSumGradV2
Oblicza gradienty dla SparseSegmentSum.
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ę> SparseSegmentSumGradV2 <T, U> | |
| Wyjście <T> | wyjście () |
| Wyjście <U> |
Metody dziedziczone
Metody publiczne
public static SparseSegmentSumGradV2 <T, U> create ( 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ę SparseSegmentSumGradV2.
Parametry
| zakres | aktualny zakres |
|---|---|
| absolwent | gradient propagowany do SparseSegmentSum op. |
| indeksy | indeksy przekazywane do odpowiedniego SparseSegmentSum op. |
| identyfikatory segmentu | segment_ids przekazane do odpowiedniej operacji SparseSegmentSum. |
| gęstyWyjścieDim0 | wymiar 0 „danych” przekazywanych do SparseSegmentSum op. |
Zwroty
- nowa instancja SparseSegmentSumGradV2