공개 최종 클래스 EnqueueTPUEmbeddingRaggedTensorBatch
tf.nn.embedding_lookup()을 사용하는 코드 포팅을 쉽게 합니다.
Sample_splits[i], embedding_indices[i] 및 aggregation_weights[i]는 i번째 특성에 해당합니다. table_ids[i]는 i번째 기능을 조회할 임베딩 테이블을 나타냅니다.
두 개의 입력 목록인 embedding_indices 및 aggregation_weights에서 해당 위치에 있는 텐서는 동일한 모양을 가져야 합니다. 즉, 해당 기능에 의해 설명된 테이블에 대한 총 조회 수와 동일한 희미한 크기()를 갖는 랭크 1을 가져야 합니다.
중첩 클래스
| 수업 | EnqueueTPUEmbeddingRaggedTensorBatch.Options | EnqueueTPUEmbeddingRaggedTensorBatch 의 선택적 속성 | |
상수
| 끈 | OP_NAME | TensorFlow 코어 엔진에서 알려진 이 작업의 이름 |
공개 방법
| 정적 EnqueueTPUEmbeddingRaggedTensorBatch.Options | 결합자 (List<String> 결합자) |
| static <T는 TNumber를 확장하고, U는 TNumber를 확장하고, V는 TNumber를 확장합니다. > EnqueueTPUEmbeddingRaggedTensorBatch | |
| 정적 EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (긴 장치Ordinal) |
| 정적 EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (List<Long> maxSequenceLengths) |
상속된 메서드
상수
공개 정적 최종 문자열 OP_NAME
TensorFlow 코어 엔진에서 알려진 이 작업의 이름
상수 값: "EnqueueTPUEmbeddingRaggedTensorBatch"
공개 방법
공개 정적 EnqueueTPUEmbeddingRaggedTensorBatch.Options 결합자 (List<String> 결합자)
매개변수
| 결합기 | 가중치 합산 후 임베딩 활성화를 정규화하는 방법을 지정하는 각 임베딩 테이블당 하나씩 문자열 스칼라 목록입니다. 지원되는 결합자는 'mean', 'sum' 또는 'sqrtn'입니다. '평균'에 대한 가중치 합이 0이 되거나 'sqrtn'에 대한 제곱 가중치의 합이 0이 되는 것은 유효하지 않습니다. 결합자가 전달되지 않으면 기본값은 모든 테이블에 'sum'을 사용하는 것입니다. |
|---|
public static EnqueueTPUEmbeddingRaggedTensorBatch create ( Scope 범위, Iterable< Operand <T>> SampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options.. . 옵션)
새로운 EnqueueTPUEmbeddingRaggedTensorBatch 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| 샘플분할 | embedding_indices 및 aggregation_weights를 행으로 분할하기 위한 중단점을 지정하는 순위 1 Tensor의 목록입니다. ids가 RaggedTensor인 경우 embedding_lookup()의 ids.row_splits에 해당합니다. |
| 임베딩인덱스 | 임베딩 테이블에 대한 인덱스인 순위 1 Tensor의 목록입니다. ids가 RaggedTensor인 경우 embedding_lookup()의 ids.values에 해당합니다. |
| 집계가중치 | 훈련 예시별 집계 가중치를 포함하는 순위 1 Tensor의 목록입니다. 이는 ids가 RaggedTensor인 경우 embedding_lookup()의 id와 동일한 row_splits를 갖는 RaggedTensor의 값 필드에 해당합니다. |
| 모드 재정의 | TPUEmbeddingConfiguration에 지정된 모드를 재정의하는 문자열 입력입니다. 지원되는 값은 {'unspecified', 'inference', 'training', 'backward_pass_only'}입니다. 'unspecified'로 설정되면 TPUEmbeddingConfiguration에 설정된 모드가 사용되고, 그렇지 않으면 mode_override가 사용됩니다. |
| 테이블 ID | 해당 입력을 조회하기 위한 임베딩 테이블의 식별자(TPUEmbeddingConfiguration의 TableDescriptor 오프셋)를 지정하는 정수 목록입니다. i번째 입력은 table_ids[i]를 사용하여 조회됩니다. table_ids 목록의 크기는 Sample_indices, embedding_indices 및 aggregation_weights의 목록 크기와 동일해야 합니다. |
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- EnqueueTPUEmbeddingRaggedTensorBatch의 새로운 인스턴스
공개 정적 EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
매개변수
| 장치서수 | 사용할 TPU 기기입니다. 0보다 크고 노드가 배치된 작업의 TPU 코어 수보다 작아야 합니다. |
|---|