NcclReduce

공개 최종 클래스 NcclReduce

단일 장치에 대한 `reduction`을 사용하여 `num_devices`에서 `input`을 줄입니다.

단일 장치에 대한 `reduction`을 사용하여 `num_devices`에서 `input`을 줄입니다.

모든 입력에 유효한 장치 할당이 있고 작업 자체에는 이러한 장치 중 하나가 할당되도록 그래프를 구성해야 합니다.

입력: 축소에 대한 입력입니다. 데이터: 모든 `num_devices` 장치에 대한 감소 값입니다. 감소: 수행할 축소 작업입니다.

공개 방법

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

상속된 메서드

공개 방법

공개 출력 <T> asOutput ()

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

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

public static NcclReduce <T> create ( Scope 범위, Iterable< Operand <T>> 입력, 문자열 감소)

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

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

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