このページは Cloud Translation API によって翻訳されました。
Switch to English

tensorflow :: ops :: バリアテイク多

#include <data_flow_ops.h>

バリアから指定された数の完了した要素を取得します。

概要

この演算は、完成した要素のコンポーネントテンソルを0次元に沿って連結して、単一のコンポーネントテンソルを作成します。

要素は、完成すると、バリアに配置された順序でバリアから出てきます。インデックス出力は、各要素が最初にバリアに挿入されたバッチに関する情報を提供します。

引数:

  • scope: Scopeオブジェクト
  • ハンドル:バリアへのハンドル。
  • num_elements:取得する要素の数を含む単一要素テンソル。
  • component_types:値の各コンポーネントのタイプ。

オプションの属性( Attrs参照):

  • allow_small_batch:バリアが既に閉じている場合、num_elements未満のアイテムを返すことができます。
  • timeout_ms:キューが空の場合、この操作は最大timeout_msミリ秒の間ブロックされます。注:このオプションはまだサポートされていません。

戻り値:

  • Outputインデックス:長さがnum_elemsのインデックスの1次元テンソル。これらのインデックスは、値がバリアに配置されたバッチを参照します(MIN_LONGで始まり、各BarrierInsertManyで増加します)。
  • Outputキー:長さがnum_elementsのキーの1次元テンソル。
  • OutputList値:バリア要素のコンポーネントごとに1つの任意次元テンソル。 すべての値は、0番目の次元の長さがnum_elementsです。

コンストラクタとデストラクタ

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)

公開属性

indices
keys
operation
values

パブリック静的関数

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

構造

tensorflow :: ops :: BarrierTakeMany :: Attrs

BarrierTakeManyのオプションの属性セッター。

公開属性

指数

::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
)

パブリック静的関数

AllowSmallBatch

Attrs AllowSmallBatch(
  bool x
)

TimeoutMs

Attrs TimeoutMs(
  int64 x
)

WaitForIncomplete

Attrs WaitForIncomplete(
  bool x
)