공개 최종 클래스 BarrierTakeMany
장벽에서 주어진 수의 완성된 요소를 가져옵니다.
이 작업은 0차원을 따라 완성된 요소 구성 요소 텐서를 연결하여 단일 구성 요소 텐서를 만듭니다.
요소는 완료되면 장벽에 배치된 순서대로 장벽에서 나옵니다. 인덱스 출력은 각 요소가 원래 장벽에 삽입된 배치에 대한 정보를 제공합니다.
중첩 클래스
수업 | BarrierTakeMany.Options | BarrierTakeMany 의 선택적 속성 |
공개 방법
정적 BarrierTakeMany.Options | 허용SmallBatch (부울 허용SmallBatch) |
정적 BarrierTakeMany | |
출력 <긴> | 인덱스 () 길이가 num_elems인 인덱스의 1차원 텐서입니다. |
출력 <문자열> | 키 () 길이가 num_elements인 1차원 키 텐서입니다. |
정적 BarrierTakeMany.Options | timeoutMs (긴 timeoutMs) |
목록< 출력 <?>> | 값 () 장벽 요소의 구성요소당 하나의 임의차원 텐서. |
정적 BarrierTakeMany.Options | waitForIncomplete (부울 waitForIncomplete) |
상속된 메서드
공개 방법
공개 정적 BarrierTakeMany.Options 허용SmallBatch (부울 허용SmallBatch)
매개변수
작은 배치 허용 | 장벽이 이미 닫힌 경우 num_elements개 미만의 항목을 반환하도록 허용합니다. |
---|
공개 정적 BarrierTakeMany 생성 ( 범위 범위, 피연산자 <String> 핸들, 피연산자 <Integer> numElements, List<Class<?>> 구성 요소 유형, 옵션... 옵션)
새로운 BarrierTakeMany 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
핸들 | 장벽의 손잡이. |
요소 수 | 가져올 요소 수를 포함하는 단일 요소 텐서입니다. |
구성 요소 유형 | 값의 각 구성 요소 유형입니다. |
옵션 | 선택적 속성 값을 전달합니다. |
보고
- BarrierTakeMany의 새로운 인스턴스
공개 출력 <Long> 인덱스 ()
길이가 num_elems인 인덱스의 1차원 텐서입니다. 이러한 인덱스는 값이 장벽에 배치된 배치를 나타냅니다(MIN_LONG으로 시작하여 각 BarrierInsertMany로 증가).
공개 정적 BarrierTakeMany.Options timeoutMs (긴 timeoutMs)
매개변수
시간 초과(Ms) | 대기열이 비어 있으면 이 작업은 최대 timeout_ms 밀리초 동안 차단됩니다. 참고: 이 옵션은 아직 지원되지 않습니다. |
---|