CollectivePermute

공개 최종 클래스 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> 입력, 피연산자 <Integer> sourceTargetPairs)
새로운 CollectivePermute 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <T>
출력 ()
순열된 입력입니다.

상속된 메서드

공개 방법

공개 출력 <T> asOutput ()

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

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

공개 정적 CollectivePermute <T> 생성 ( 범위 범위, 피연산자 <T> 입력, 피연산자 <Integer> sourceTargetPairs)

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

매개변수
범위 현재 범위
입력 순열될 로컬 입력입니다. 현재는 float 및 bfloat16만 지원합니다.
sourceTargetPairs [num_pairs, 2] 형태의 텐서.
보고
  • CollectivePermute의 새로운 인스턴스

공개 출력 <T> 출력 ()

순열된 입력입니다.