ML Topluluk Günü 9 Kasım! TensorFlow, JAX güncellemeler için bize katılın ve daha fazla bilgi edinin

tensorflow :: ops :: SıralanmamışSegmentSum

#include <math_ops.h>

Toplamı bir tensörün segmentleri boyunca hesaplar.

Özet

Segmentlerin açıklaması için segmentasyon bölümünü okuyun.

\(output[i] = {j...} data[j...]\), toplamın tuples j... üzerinde olduğu şekilde bir tensör hesaplar, öyle ki segment_ids[j...] == i . SegmentSum farklı SegmentSum , segment_ids sıralanması gerekmez ve tüm geçerli değerler aralığındaki tüm değerleri kapsaması gerekmez.

Toplam, belirli bir segment kimliği i için boşsa, output[i] = 0 . Verilen segment kimliği i negatifse, değer çıkarılır ve segmentin toplamına eklenmez.

num_segments , farklı segment kimliklerinin sayısına eşit olmalıdır.

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]]

Argümanlar:

  • kapsam: Bir Scope nesnesi
  • segment_ids: Şekli data.shape öneki olan bir tensör.

İadeler:

  • Output : boyut num_segments içeren tek bir boyutla değiştirilen ilk segment_ids.rank boyutları dışında verilerle aynı şekle sahiptir.

Yapıcılar ve Yıkıcılar

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

Genel özellikler

operation
output

Kamusal işlevler

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Genel özellikler

operasyon

Operation operation

çıktı

::tensorflow::Output output

Kamusal işlevler

SıralanmamışSegmentSum

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

düğüm

::tensorflow::Node * node() const 

operator :: tensorflow :: Girdi

 operator::tensorflow::Input() const 

operator :: tensorflow :: Çıktı

 operator::tensorflow::Output() const