tensör akışı:: operasyon:: BariyerBirçok Al

Bir bariyerden verilen sayıda tamamlanmış öğeyi alır.

Özet

Bu işlem, tek bileşenli bir tensör yapmak için 0. boyut boyunca tamamlanmış eleman bileşen tensörlerini birleştirir.

Öğeler, tamamlandıklarında ve bariyere yerleştirildikleri sırayla bariyerden çıkarlar. Endeks çıktısı, her bir elemanın orijinal olarak bariyere eklendiği yığın hakkında bilgi sağlar.

Argümanlar:

• kapsam: Bir Kapsam nesnesi
• tutamak: Bir engelin tutacağı.
• num_elements: Alınacak eleman sayısını içeren tek elemanlı bir tensör.
• component_types: Bir değerdeki her bileşenin türü.

İsteğe bağlı özellikler (bkz. Attrs ):

• allow_small_batch: Bariyer zaten kapalıysa, num_elements öğesinden daha az öğe döndürmeye izin verir.
• timeout_ms: Kuyruk boşsa, bu işlem timeout_ms milisaniyeye kadar engellenir. Not: Bu seçenek henüz desteklenmemektedir.

İadeler:

• Output indeksleri: Uzunluğu num_elems olan tek boyutlu bir indeks tensörü. Bu endeksler, değerlerin bariyere yerleştirildiği seriye atıfta bulunur (MIN_LONG ile başlar ve her BarrierInsertMany ile artar).
• Output anahtarları: Uzunluğu num_elements olan tek boyutlu bir anahtar tensörü.
• OutputList değerleri: Bir bariyer elemanındaki bileşen başına herhangi bir boyutlu bir tensör. Tüm değerler, 0. boyutta uzunluk num_elements'a sahiptir.

Genel özellikler

::tensorflow::Output indices

::tensorflow::Output keys

Operation operation

::tensorflow::OutputList values

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types
)

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types,
  const BarrierTakeMany::Attrs & attrs
)

Attrs AllowSmallBatch(
  bool x
)

Attrs TimeoutMs(
  int64 x
)

Attrs WaitForIncomplete(
  bool x
)