tensorflow :: ops :: UnsortedSegmentSum
#include <math_ops.h>
Oblicza sumę wzdłuż segmentów tensora.
Podsumowanie
Przeczytaj sekcję dotyczącą segmentacji, aby uzyskać wyjaśnienie dotyczące segmentów.
Oblicza tensor taki, że \(output[i] = {j...} data[j...]\), gdzie suma jest powyżej krotek j...
taki, że segment_ids[j...] == i
. W przeciwieństwie SegmentSum
, segment_ids
nie muszą być sortowane i nie musi obejmować wszystkie wartości w pełnym zakresie prawidłowych wartości.
Jeśli suma jest pusta dla danego identyfikatora segmentu i
, output[i] = 0
. Jeśli podany identyfikator segmentu i
jest ujemny, wartość jest odrzucana i nie zostanie dodana do sumy segmentu.
num_segments
powinna być równa liczbie różnych identyfikatorów segmentów.
c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]]) tf.unsorted_segment_sum(c, tf.constant([0, 1, 0]), num_segments=2) # ==> [[ 5, 5, 5, 5], # [5, 6, 7, 8]]
Argumenty:
- zakres: obiekt Scope
- segment_ids: tensor, którego kształt jest prefiksem
data.shape
.
Zwroty:
-
Output
: Ma taki sam kształt jak dane, z wyjątkiem pierwszychsegment_ids.rank
wymiarach, które zostały zastąpione jednym wymiarze, który ma rozmiarnum_segments
.
Konstruktorzy i niszczyciele | |
---|---|
UnsortedSegmentSum (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids, :: tensorflow::Input num_segments) |
Atrybuty publiczne | |
---|---|
operation | |
output |
Funkcje publiczne | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Atrybuty publiczne
operacja
Operation operation
wynik
::tensorflow::Output output
Funkcje publiczne
UnsortedSegmentSum
UnsortedSegmentSum( const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input segment_ids, ::tensorflow::Input num_segments )
węzeł
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Output
operator::tensorflow::Output() const