공개 최종 클래스 AllReduce
동일한 유형과 모양의 여러 텐서를 상호 축소합니다.
중첩 클래스
| 수업 | AllReduce.옵션 | AllReduce 의 선택적 속성 | |
상수
| 끈 | OP_NAME | TensorFlow 코어 엔진에서 알려진 이 작업의 이름 |
공개 방법
| 출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
| 정적 AllReduce.Options | 커뮤니케이션힌트 (문자열 커뮤니케이션힌트) |
| static <T는 TNumber를 확장합니다. > AllReduce <T> | 생성 ( 범위 범위, 피연산자 <T> 입력, Long groupSize, Long groupKey, Long instanceKey, String mergeOp, String finalOp, List<Long> subdivOffsets, Options... 옵션) 새로운 AllReduce 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다. |
| 출력 <T> | 데이터 () |
| 정적 AllReduce.Options | timeoutSeconds (부동 timeoutSeconds) |
| 정적 AllReduce.Options | waitFor (List<Long> 대기) |
상속된 메서드
상수
공개 정적 최종 문자열 OP_NAME
TensorFlow 코어 엔진에서 알려진 이 작업의 이름
상수 값: "CollectiveReduce"
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
공개 정적 AllReduce <T> 생성 ( 범위 범위, 피연산자 <T> 입력, 긴 groupSize, 긴 groupKey, 긴 인스턴스 키, 문자열 mergeOp, 문자열 finalOp, List<Long> subdivOffsets, 옵션... 옵션)
새로운 AllReduce 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- AllReduce의 새로운 인스턴스