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