공개 최종 클래스 EnqueueTPUEmbeddingBatch
입력 배치 텐서 목록을 TPUEmbedding에 추가하는 작업입니다.
입력 배치 텐서 목록을 TPUEmbedding에 추가하는 작업입니다.
중첩 클래스
| 수업 | EnqueueTPUEmbeddingBatch.Options | EnqueueTPUEmbeddingBatch 의 선택적 속성 | |
공개 방법
| 정적 EnqueueTPUEmbeddingBatch.Options | 결합자 (List<String> 결합자) |
| 정적 EnqueueTPUEmbeddingBatch | |
| 정적 EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (긴 장치Ordinal) |
상속된 메서드
공개 방법
공개 정적 EnqueueTPUEmbeddingBatch.Options 결합자 (List<String> 결합자)
매개변수
| 결합기 | 가중치 합산 후 임베딩 활성화를 정규화하는 방법을 지정하는 각 임베딩 테이블당 하나씩 문자열 스칼라 목록입니다. 지원되는 결합자는 'mean', 'sum' 또는 'sqrtn'입니다. '평균'에 대한 가중치 합이 0이 되거나 'sqrtn'에 대한 제곱 가중치의 합이 0이 되는 것은 유효하지 않습니다. 결합자가 전달되지 않으면 기본값은 모든 테이블에 'sum'을 사용하는 것입니다. |
|---|
공개 정적 EnqueueTPUEmbeddingBatch 생성 ( 범위 범위, Iterable< Operand <String>> 배치, Operand <String> modeOverride, 옵션... 옵션)
새로운 EnqueueTPUEmbeddingBatch 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| 일괄 | dist_belief.SparseFeatures protos로 인코딩된 배치 입력을 포함하는 각 임베딩 테이블당 하나씩 1D 텐서 목록입니다. SparseFeatures proto의 가중치 필드가 ID에 대해 채워지지 않으면 가중치 1.0이 가정됩니다. |
| 모드 재정의 | TPUEmbeddingConfiguration에 지정된 모드를 재정의하는 문자열 입력입니다. 지원되는 값은 {'unspecified', 'inference', 'training', 'backward_pass_only'}입니다. 'unspecified'로 설정되면 TPUEmbeddingConfiguration에 설정된 모드가 사용되고, 그렇지 않으면 mode_override가 사용됩니다. |
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- EnqueueTPUEmbeddingBatch의 새 인스턴스
공개 정적 EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)
매개변수
| 장치서수 | 사용할 TPU 기기입니다. Op가 TPU 기기에서 실행 중인 경우 -1이어야 하고, Op가 CPU 기기에서 실행 중인 경우 >= 0이어야 합니다. |
|---|