공개 최종 클래스 TPUPartitionedOutputV2
XLA에 의해 분할될 텐서를 분할된 목록으로 역다중화하는 작업입니다.
XLA 계산 외부로 출력됩니다. ND 샤딩을 지원합니다.
공개 방법
정적 <T> TPUPartitionedOutputV2 <T> | |
반복자< 피연산자 <T>> | 반복자 () |
목록< 출력 <T>> | 출력 () 동일한 모양을 가진 분할된 출력 목록입니다. |
상속된 메서드
공개 방법
공개 정적 TPUPartitionedOutputV2 <T> 생성 ( 범위 범위, 피연산자 <T> 입력, Long numSplits, List<Long> partitionDims)
새 TPUPartitionedOutputV2 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
입력 | 분할된 텐서의 전체 형태를 나타내는 텐서입니다. |
partitionDim | 각 차원이 분할되는 방식을 설명하는 정수 목록입니다. 비어 있으면 입력이 복제되었음을 나타냅니다. |
보고
- TPUPartitionedOutputV2의 새 인스턴스