텐서플로우:: 작전:: 장벽가져다많은

#include <data_flow_ops.h>

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

요약

이 작업은 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입니다.

생성자와 소멸자

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:: 속성

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
)

공개 정적 함수

작은 배치 허용

Attrs AllowSmallBatch(
  bool x
)

시간 초과(Ms)

Attrs TimeoutMs(
  int64 x
)

미완료 대기

Attrs WaitForIncomplete(
  bool x
)