공개 최종 클래스 CollectivePermute
복제된 TPU 인스턴스 전체에서 텐서를 순열하는 작업입니다.
각 인스턴스는 자체 입력을 제공합니다.
예를 들어 `[A, B, C, D]`라는 4개의 TPU 인스턴스가 있다고 가정합니다. source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]`을 전달하면 `[D, A, B, C]`가 출력됩니다.
공개 방법
출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
정적 <T> CollectivePermute <T> | |
출력 <T> | 출력 () 순열된 입력입니다. |
상속된 메서드
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
공개 정적 CollectivePermute <T> 생성 ( 범위 범위, 피연산자 <T> 입력, 피연산자 <Integer> sourceTargetPairs)
새로운 CollectivePermute 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
입력 | 순열될 로컬 입력입니다. 현재는 float 및 bfloat16만 지원합니다. |
sourceTargetPairs | [num_pairs, 2] 형태의 텐서. |
보고
- CollectivePermute의 새로운 인스턴스