przepływ tensorowy:: ops:: Nieposortowana suma segmentu

Oblicza sumę wzdłuż segmentów tensora.

Streszczenie

Przeczytaj sekcję dotyczącą segmentacji, aby uzyskać wyjaśnienie segmentów.

Oblicza tensor taki, że \(output[i] = {j...} data[j...]\) gdzie suma jest po krotkach j... tak, że segment_ids[j...] == i . W przeciwieństwie do SegmentSum , segment_ids nie musi być sortowany i nie musi obejmować wszystkich wartości w pełnym zakresie prawidłowych wartości.

Jeżeli suma jest pusta dla danego identyfikatora segmentu i , output[i] = 0 . Jeżeli podany identyfikator segmentu i jest ujemny, wartość jest odrzucana i nie będzie dodawana do sumy segmentu.

num_segments powinna być równa liczbie odrębnych 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 przedrostkiem data.shape .

Zwroty:

• Output : mają 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 .

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