NcclAllReduce

공개 최종 클래스 NcclAllReduce

모든 입력 텐서에 대한 감소를 포함하는 텐서를 출력합니다.

동일한 `shared_name 내에서 ops에 전달된 모든 입력 텐서에 대한 감소를 포함하는 텐서를 출력합니다.

하나의 작업이 shared_name 값 `c`로 실행되면 `num_devices` 작업이 shared_name 값 `c`로 실행되도록 그래프를 구성해야 합니다. 그렇지 않으면 그래프 실행이 완료되지 않습니다.

입력: 감소 데이터에 대한 입력: 모든 `num_devices` 장치에 대한 감소 값입니다. 감소: 수행할 축소 작업입니다. num_devices: 이 감소에 참여하는 장치 수입니다. shared_name: 동일한 감소 작업 간에 공유되는 식별자입니다.

공개 방법

출력 <T>
출력 ()
텐서의 기호 핸들을 반환합니다.
static <T 확장 번호> NcclAllReduce <T>
생성 ( 범위 범위, 피연산자 <T> 입력, 문자열 감소, Long numDevices, 문자열 sharedName)
새로운 NcclAllReduce 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <T>

상속된 메서드

공개 방법

공개 출력 <T> asOutput ()

텐서의 기호 핸들을 반환합니다.

TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.

public static NcclAllReduce <T> create ( 범위 범위, 피연산자 <T> 입력, 문자열 감소, Long numDevices, 문자열 sharedName)

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

매개변수
범위 현재 범위
보고
  • NcclAllReduce의 새로운 인스턴스

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