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

tensorflow :: ops :: BarrierTakeMany

#include <data_flow_ops.h>

Belirli sayıda tamamlanmış elemanı bir bariyerden alır.

Özet

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

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

Argümanlar:

  • kapsam: Bir Scope nesnesi
  • tutacak: Bir bariyerin tutacağı.
  • num_elements: Alınacak eleman sayısını içeren tek elemanlı bir tensör.
  • bileşen_türleri: 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ının döndürülmesine izin verin.
  • timeout_ms: Kuyruk boşsa, bu işlem timeout_ms milisaniyeye kadar engelleyecektir. Not: Bu seçenek henüz desteklenmemektedir.

İadeler:

  • Output indisleri: Uzunluk sayı_emleri olan tek boyutlu bir indeks tensörü. Bu endeksler, değerlerin bariyere yerleştirildiği partiye karşılık gelir (MIN_LONG ile başlar ve her BarrierInsertMany ile artar ).
  • Output anahtarları: Uzunluk num_elements ile tek boyutlu bir anahtar tensörü.
  • OutputList değerleri: Bir bariyer elemanındaki bileşen başına herhangi bir boyutlu tensör. Tüm değerlerin 0. boyutta uzunluk sayıları vardır.

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

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)

Genel özellikler

indices
keys
operation
values

Genel statik işlevler

AllowSmallBatch (bool x)
TimeoutMs (int64 x)
WaitForIncomplete (bool x)

Yapılar

tensorflow :: ops :: BarrierTakeMany :: Attrs

BarrierTakeMany için isteğe bağlı öznitelik belirleyiciler.

Genel özellikler

endeksler

::tensorflow::Output indices

anahtarlar

::tensorflow::Output keys

operasyon

Operation operation

değerler

::tensorflow::OutputList values

Kamusal işlevler

BarrierTakeMany

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

BarrierTakeMany

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

Genel statik işlevler

AllowSmallBatch

Attrs AllowSmallBatch(
  bool x
)

Zaman aşımı

Attrs TimeoutMs(
  int64 x
)

WaitForIncomplete

Attrs WaitForIncomplete(
  bool x
)