ExperimentalAutoShardDataset

공개 최종 클래스 ExperimentalAutoShardDataset

입력 데이터 세트를 샤딩하는 데이터 세트를 생성합니다.

num_workers별로 입력 데이터 세트를 샤딩하고 인덱스 번째 작업자에 대해 샤딩된 데이터 세트를 반환하는 데이터 세트를 생성합니다. 이는 데이터 세트 그래프를 검사하고 리더 데이터 세트(예: CSVDataset, TFRecordDataset)에 대한 입력 앞에 샤드 작업을 삽입하여 데이터 세트를 자동으로 샤딩하려고 시도합니다.

데이터 세트를 자동으로 샤딩할 수 없는 경우 이 데이터 세트는 NotFound 오류를 발생시킵니다.

중첩 클래스

수업 ExperimentalAutoShardDataset.Options ExperimentalAutoShardDataset 의 선택적 속성

공개 방법

<객체> 출력
출력 ()
텐서의 기호 핸들을 반환합니다.
정적 ExperimentalAutoShardDataset.Options
autoShardPolicy (긴 autoShardPolicy)
정적 ExperimentalAutoShardDataset
생성 ( Scope 범위, Operand <?> inputDataset, Operand <Long> numWorkers, Operand <Long> index, List<Class<?>> outputTypes, List< Shape > outputShapes, Options... 옵션)
새로운 ExperimentalAutoShardDataset 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <?>
핸들 ()

상속된 메서드

공개 방법

공개 출력 <Object> asOutput ()

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

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

공개 정적 ExperimentalAutoShardDataset.Options autoShardPolicy (Long autoShardPolicy)

public static ExperimentalAutoShardDataset create ( Scope 범위, Operand <?> inputDataset, Operand <Long> numWorkers, Operand <Long> index, List<Class<?>> outputTypes, List< Shape > outputShapes, Options... 옵션)

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

매개변수
범위 현재 범위
입력데이터세트 입력 데이터 세트를 나타내는 변형 텐서입니다.
작업자 수 이 데이터 세트를 배포할 작업자 수를 나타내는 스칼라입니다.
색인 num_workers 중 현재 작업자의 인덱스를 나타내는 스칼라입니다.
옵션 선택적 속성 값을 전달합니다.
보고
  • ExperimentalAutoShardDataset의 새 인스턴스

공개 출력 <?> 핸들 ()