TPUPartitionedOutputV2

공개 최종 클래스 TPUPartitionedOutputV2

XLA에 의해 분할될 텐서를 분할된 목록으로 역다중화하는 작업입니다.

XLA 계산 ​​외부로 출력됩니다. ND 샤딩을 지원합니다.

공개 방법

정적 <T> TPUPartitionedOutputV2 <T>	생성 ( 범위 범위, 피연산자 <T> 입력, Long numSplits, List<Long> partitionDims) 새 TPUPartitionedOutputV2 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
반복자< 피연산자 <T>>	반복자 ()
목록< 출력 <T>>	출력 () 동일한 모양을 가진 분할된 출력 목록입니다.

상속된 메서드

org.tensorflow.op.PrimitiveOp 클래스에서

최종 부울	같음 (객체 객체)
최종 정수	해시 코드 ()
작업	op () 기본 Operation 반환합니다.
최종 문자열	toString ()

java.lang.Object 클래스에서

부울	같음 (개체 arg0)
마지막 수업<?>	getClass ()
정수	해시 코드 ()
최종 무효	알림 ()
최종 무효	통지모두 ()
끈	toString ()
최종 무효	대기 (long arg0, int arg1)
최종 무효	기다리세요 (긴 arg0)
최종 무효	기다리다 ()

인터페이스 java.lang.Iterable에서

무효의	forEach (소비자<? super T> arg0)
추상 반복자< 피연산자 <T>>	반복자 ()
분할기< 피연산자 <T>>	분할기 ()

공개 방법

공개 정적 TPUPartitionedOutputV2 <T> 생성 ( 범위 범위, 피연산자 <T> 입력, Long numSplits, List<Long> partitionDims)

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

매개변수

범위	현재 범위
입력	분할된 텐서의 전체 형태를 나타내는 텐서입니다.
partitionDim	각 차원이 분할되는 방식을 설명하는 정수 목록입니다. 비어 있으면 입력이 복제되었음을 나타냅니다.

보고

• TPUPartitionedOutputV2의 새 인스턴스

공개 반복자< 피연산자 <T>> 반복자 ()

공개 목록< 출력 <T>> 출력 ()

동일한 모양을 가진 분할된 출력 목록입니다.