텐서플로:: 작전:: Barrier TakeMany

장벽에서 주어진 수의 완료된 요소를 가져옵니다.

요약

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

요소는 완료될 때 그리고 장벽에 배치된 순서대로 장벽에서 나옵니다. 인덱스 출력은 각 요소가 원래 장벽에 삽입된 배치에 대한 정보를 제공합니다.

인수:

• 범위: 범위 개체
• 핸들: 장벽에 대한 핸들입니다.
• 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
)