AllReduce

공개 최종 클래스 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.Options 통신Hint (문자열 통신Hint)

공개 정적 AllReduce <T> 생성 ( 범위 범위, 피연산자 <T> 입력, 긴 groupSize, 긴 groupKey, 긴 인스턴스 키, 문자열 mergeOp, 문자열 finalOp, List<Long> subdivOffsets, 옵션... 옵션)

새로운 AllReduce 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
옵션 선택적 속성 값을 전달합니다.
보고
  • AllReduce의 새로운 인스턴스

공개 출력 <T> 데이터 ()

공개 정적 AllReduce.Options timeoutSeconds (부동 timeoutSeconds)

공개 정적 AllReduce.Options waitFor (List<Long> waitFor)