tensorflow :: ops :: UnsortedSegmentSum

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 pierwszych segment_ids.rank wymiarach, które zostały zastąpione jednym wymiarze, który ma rozmiar num_segments .

Atrybuty publiczne

Operation operation

::tensorflow::Output output

 UnsortedSegmentSum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input segment_ids,
  ::tensorflow::Input num_segments
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const