공개 최종 클래스 NcclReduce
단일 장치에 대한 `reduction`을 사용하여 `num_devices`에서 `input`을 줄입니다.
단일 장치에 대한 `reduction`을 사용하여 `num_devices`에서 `input`을 줄입니다.
모든 입력에 유효한 장치 할당이 있고 작업 자체에는 이러한 장치 중 하나가 할당되도록 그래프를 구성해야 합니다.
입력: 축소에 대한 입력입니다. 데이터: 모든 `num_devices` 장치에 대한 감소 값입니다. 감소: 수행할 축소 작업입니다.
상수
끈 | OP_NAME | TensorFlow 코어 엔진에서 알려진 이 작업의 이름 |
공개 방법
출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
static <T는 TNumber를 확장합니다. > NcclReduce <T> | |
출력 <T> | 데이터 () |
상속된 메서드
상수
공개 정적 최종 문자열 OP_NAME
TensorFlow 코어 엔진에서 알려진 이 작업의 이름
상수 값: "NcclReduce"
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
public static NcclReduce <T> create ( Scope 범위, Iterable< Operand <T>> 입력, 문자열 감소)
새로운 NcclReduce 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|
보고
- NcclReduce의 새로운 인스턴스