Oblicza sumę wzdłuż rzadkich segmentów tensora.
Przeczytaj [sekcję o segmentacji](https://tensorflow.org/api_docs/python/tf/math#Segmentation), aby uzyskać wyjaśnienie segmentów.
Podobnie jak „SegmentSum”, ale „id_segmentu” może mieć rangę mniejszą niż pierwszy wymiar „danych”, wybierając podzbiór wymiaru 0, określony przez „indeksy”.
Na przykład:
c = tf.constant([[1,2,3,4], [-1,-2,-3,-4], [5,6,7,8]])
# Select two rows, one segment.
tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0]))
# => [[0 0 0 0]]
# Select two rows, two segment.
tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1]))
# => [[ 1 2 3 4]
# [-1 -2 -3 -4]]
# Select all rows, two segments.
tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1]))
# => [[0 0 0 0]
# [5 6 7 8]]
# Which is equivalent to:
tf.segment_sum(c, tf.constant([0, 0, 1]))
Stałe
Smyczkowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
statyczny <T rozszerza TNumber > SparseSegmentSum <T> | |
Wyjście <T> | wyjście () Ma taki sam kształt jak dane, z wyjątkiem wymiaru 0, który ma rozmiar „k”, czyli liczbę segmentów. |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
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 SparseSegmentSum <T> utwórz (zakres zakresu , dane argumentu <T>, argument <? rozszerza TNumber > indeksy, argument <? rozszerza TNumber > identyfikatory segmentów)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SparseSegmentSum.
Parametry
zakres | aktualny zakres |
---|---|
indeksy | Tensor 1-D. Ma tę samą rangę co `segment_ids`. |
identyfikatory segmentu | Tensor 1-D. Wartości należy posortować i można je powtarzać. |
Powroty
- nowa instancja SparseSegmentSum
publiczne wyjście <T> wyjście ()
Ma taki sam kształt jak dane, z wyjątkiem wymiaru 0, który ma rozmiar „k”, czyli liczbę segmentów.