TPUPartitionedOutput

공개 최종 클래스 TPUPartitionedOutput

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

XLA 계산 ​​외부로 출력됩니다.

중첩 클래스

수업 TPUPartitionedOutput.옵션 TPUPartitionedOutput 의 선택적 속성

공개 방법

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

상속된 메서드

공개 방법

공개 정적 TPUPartitionedOutput <T> 생성 ( 범위 범위, 피연산자 <T> 입력, Long numSplits, 옵션... 옵션)

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

매개변수
범위 현재 범위
입력 분할된 텐서의 전체 형태를 나타내는 텐서입니다.
옵션 선택적 속성 값을 전달합니다.
보고
  • TPUPartitionedOutput의 새 인스턴스

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

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

동일한 모양을 가져야 하는 분할된 입력 목록입니다.

공개 정적 TPUPartitionedOutput.Options partitionDim (Long partitionDim)

매개변수
파티션딤 정수는 분할된 차원을 설명합니다.