Oblicza sumę wzdłuż segmentów tensora.
Przeczytaj [sekcję o segmentacji](https://tensorflow.org/api_docs/python/tf/math#Segmentation), aby uzyskać wyjaśnienie segmentów.
Oblicza tensor taki, że \\(output_i = \sum_j data_j\\) gdzie suma jest większa od `j` tak, że `segment_ids[j] == i`.
Jeżeli suma jest pusta dla danego segmentu o identyfikatorze `i`, `output[i] = 0`.
Pamiętaj, że ta opcja jest obecnie obsługiwana tylko z jit_compile=True.
Metody publiczne
Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
statyczny <T, U rozszerza liczbę, V rozszerza liczbę> SegmentSumV2 <T> | |
Wyjście <T> | wyjście () Ma taki sam kształt jak dane, z wyjątkiem pierwszych wymiarów `segment_ids.rank`, które są zastępowane pojedynczym wymiarem o rozmiarze `num_segments`. |
Metody dziedziczone
Metody publiczne
publiczne wyjście <T> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static SegmentSumV2 <T> utwórz (zakres zakresu , dane argumentu <T>, argument <U> segmentIds, argument <V> numSegments)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SegmentSumV2.
Parametry
zakres | aktualny zakres |
---|---|
identyfikatory segmentu | Tensor 1-D, którego rozmiar jest równy rozmiarowi pierwszego wymiaru „danych”. Wartości należy posortować i można je powtarzać. Wartości muszą być mniejsze niż „liczba_segmentów”. Uwaga: Wartości są zawsze sprawdzane pod kątem sortowania na procesorze, nigdy na GPU. |
Powroty
- nowa instancja SegmentSumV2
publiczne wyjście <T> wyjście ()
Ma taki sam kształt jak dane, z wyjątkiem pierwszych wymiarów `segment_ids.rank`, które są zastępowane pojedynczym wymiarem o rozmiarze `num_segments`.