CrossReplicaSum

공개 최종 클래스 CrossReplicaSum

복제된 TPU 인스턴스 전체에서 입력을 합산하는 작업입니다.

각 인스턴스는 자체 입력을 제공합니다.

예를 들어 `[A, B, C, D, E, F, G, H]`라는 8개의 TPU 인스턴스가 있다고 가정합니다. group_location=`[[0,2,4,6],[1,3,5,7]]`을 전달하면 `A, C, E, G`를 그룹 0으로 설정하고 `B, D, F, H`를 설정합니다. 그룹 1로. 따라서 출력은 다음과 같습니다: `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ G, B+D+F+H, A+C+E+G, B+D+F+H]`.

상수

끈

OP_NAME

TensorFlow 코어 엔진에서 알려진 이 작업의 이름

공개 방법

출력 <T>	출력 () 텐서의 기호 핸들을 반환합니다.
static <T는 TNumber를 확장합니다. > CrossReplicaSum <T>	생성 ( 범위 범위, 피연산자 <T> 입력, 피연산자 < TInt32 > groupAssignment) 새로운 CrossReplicaSum 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <T>	출력 () 모든 분산 입력의 합계입니다.

상속된 메서드

org.tensorflow.op.RawOp 클래스에서

최종 부울	같음 (객체 객체)
최종 정수	해시 코드 ()
작업	op () 이 계산 단위를 단일 Operation 으로 반환합니다.
최종 문자열	toString ()

java.lang.Object 클래스에서

부울	같음 (개체 arg0)
마지막 수업<?>	getClass ()
정수	해시 코드 ()
최종 무효	알림 ()
최종 무효	통지모두 ()
끈	toString ()
최종 무효	대기 (long arg0, int arg1)
최종 무효	기다리세요 (긴 arg0)
최종 무효	기다리다 ()

인터페이스 org.tensorflow.op.Op 에서

추상 실행환경	환경 () 이 작업이 생성된 실행 환경을 반환합니다.
추상적인 작업	op () 이 계산 단위를 단일 Operation 으로 반환합니다.

인터페이스 org.tensorflow.Operand 에서

추상 출력 <T>	출력 () 텐서의 기호 핸들을 반환합니다.
추상 T	텐서 () 이 피연산자의 텐서를 반환합니다.
추상 모양	모양 () 이 피연산자의 Output 에서 ​​참조하는 텐서의 (아마도 부분적으로 알려진) 모양을 반환합니다.
추상 클래스<T>	유형 () 이 피연산자의 텐서 유형을 반환합니다.

인터페이스 org.tensorflow.ndarray.Shaped 에서

추상 정수	순위 ()
추상 모양	모양 ()
추상적인 긴	사이즈 () 이 컨테이너의 전체 크기를 값 개수로 계산하고 반환합니다.

상수

공개 정적 최종 문자열 OP_NAME

TensorFlow 코어 엔진에서 알려진 이 작업의 이름

상수 값: "CrossReplicaSum"

공개 방법

공개 출력 <T> asOutput ()

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

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

공개 정적 CrossReplicaSum <T> 생성 ( 범위 범위, 피연산자 <T> 입력, 피연산자 < TInt32 > groupAssignment)

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

매개변수

범위	현재 범위
입력	합계에 대한 로컬 입력입니다.
그룹할당	[num_groups, num_replicas_per_group] 형태의 int32 텐서. `group_location[i]`는 i번째 하위 그룹의 복제본 ID를 나타냅니다.

보고

• CrossReplicaSum의 새 인스턴스

공개 출력 <T> 출력 ()

모든 분산 입력의 합계입니다.