공개 최종 클래스 TPUReplicatedOutput
N 방향으로 복제된 TPU 계산의 N 출력을 연결합니다.
이 작업은 `tpu.replicate()` 계산 하위 그래프의 복제된 출력을 보유합니다. 복제된 각 출력은 입력과 동일한 모양과 유형을 갖습니다.
예를 들어:
%computation = "tf.Computation"()
%replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
위 계산에는 두 복제본의 복제된 출력이 있습니다. 공개 방법
정적 <T> TPUReplicatedOutput <T> | |
반복자< 피연산자 <T>> | 반복자 () |
목록< 출력 <T>> | 출력 () |
상속된 메서드
공개 방법
공개 정적 TPUReplicatedOutput <T> 생성 ( 범위 범위, 피연산자 <T> 입력, Long numReplicas)
새 TPUReplicatedOutput 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|
보고
- TPUReplicatedOutput의 새 인스턴스