이 페이지는 Cloud Translation API를 통해 번역되었습니다.
Switch to English

tf.distribute.InputContext

TensorFlow 1 개 버전 GitHub의에서 소스보기

입력 기능에 의해 필요한 수준의 배치 정보.

이것은 사용자의 입력 함수에 전달하고, 컴퓨팅 복제 및 입력 파이프 라인에 대한 정보를 포함하는 컨텍스트 클래스입니다. (동기 훈련) 컴퓨팅 복제본의 수는 각 복제에 대해 원하는 글로벌 배치 사이즈에서 로컬 배치 크기를 계산하는 데 도움이됩니다. 입력 정보는 파이프 라인 (등 다른 입력 소스를 사용하는 예에 대한 입력을 배관 샤드) 각각의 복제본에서의 입력의 다른 일부를 반환 할 수있다.

num_input_pipelines 클러스터의 입력 파이프 라인의 수.
input_pipeline_id 현재 입력 파이프 이드 [0의 int되어야 num_input_pipelines ).
num_replicas_in_sync 동기화되어 복제본 수.

input_pipeline_id 입력 파이프 라인 ID를 돌려줍니다.
num_input_pipelines 입력 파이프 라인의 수를 돌려줍니다.
num_replicas_in_sync 반환 연산의 수는 동기화 복제본.

행동 양식

get_per_replica_batch_size

소스보기

당 복제 배치 사이즈를 돌려줍니다.

인수
global_batch_size 로 나눌 수 있어야 글로벌 배치 크기 num_replicas_in_sync .

보고
당 복제 배치 크기.

제기
ValueError 경우 global_batch_size 로 나누어되지 num_replicas_in_sync .