텐서플로우:: ops:: BarrierTakeMany

지정된 개수만큼 완료된 요소를 장벽에서 제거합니다.

요약

이 연산은 완성된 요소 구성 요소 텐서를 0번째 차원을 따라 연결하여 단일 구성 요소 텐서를 만듭니다.

요소들은 작업이 완료되면 장벽에 투입된 순서대로 장벽 밖으로 나옵니다. 인덱스 출력은 각 요소가 원래 장벽에 투입된 배치에 대한 정보를 제공합니다.

논거:

• 범위: 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 값: 장벽 요소의 각 구성 요소당 하나의 임의 차원 텐서가 출력됩니다. 모든 값은 0번째 차원에 num_elements 길이를 가집니다.

공개 속성

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