공개 최종 클래스 EnqueueTPUEmbeddingArbitraryTensorBatch
tf.nn.embedding_lookup_sparse()를 사용하는 코드 포팅을 쉽게 합니다.
embedding_indices[i] 및 aggregation_weights[i]는 i번째 특성에 해당합니다.
세 개의 입력 목록(sample_indices, embedding_indices 및 aggregation_weights)에서 해당 위치에 있는 텐서는 동일한 모양을 가져야 합니다. 즉, 해당 기능에 의해 설명된 테이블에 대한 총 조회 수와 동일한 희미한 크기()를 갖는 순위 1을 가져야 합니다.
중첩 클래스
| 수업 | EnqueueTPUEmbeddingArbitraryTensorBatch.Options | EnqueueTPUEmbeddingArbitraryTensorBatch 의 선택적 속성 | |
공개 방법
| 정적 EnqueueTPUEmbeddingArbitraryTensorBatch.Options | 결합자 (List<String> 결합자) |
| static <T는 숫자 확장, U는 숫자 확장, V는 숫자 확장> EnqueueTPUEmbeddingArbitraryTensorBatch | |
| 정적 EnqueueTPUEmbeddingArbitraryTensorBatch.Options | deviceOrdinal (긴 장치Ordinal) |
상속된 메서드
공개 방법
공개 정적 EnqueueTPUEmbeddingArbitraryTensorBatch.Options 결합자 (List<String> 결합자)
매개변수
| 결합기 | 가중치 합산 후 임베딩 활성화를 정규화하는 방법을 지정하는 각 임베딩 테이블당 하나씩 문자열 스칼라 목록입니다. 지원되는 결합자는 'mean', 'sum' 또는 'sqrtn'입니다. '평균'에 대한 가중치 합이 0이 되거나 'sqrtn'에 대한 제곱 가중치의 합이 0이 되는 것은 유효하지 않습니다. 결합자가 전달되지 않으면 기본값은 모든 테이블에 'sum'을 사용하는 것입니다. |
|---|
public static EnqueueTPUEmbeddingArbitraryTensorBatch create ( Scope 범위, Iterable< Operand <T>> SampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... 옵션)
새로운 EnqueueTPUEmbeddingArbitraryTensorBatch 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| SampleIndicesOrRowSplits | 해당 embedding_indices 및 aggregation_weights 값이 속하는 훈련 예제를 지정하는 순위 2 Tensor의 목록입니다. 첫 번째 차원의 크기가 0이면 각 embedding_indices가 다른 샘플에 속한다고 가정합니다. int32와 int64가 모두 허용되며 내부적으로 int32로 변환됩니다. 또는 embedding_indices 및 aggregation_weights를 행으로 분할하기 위해 행 분할을 지정하는 순위 1 Tensor 목록입니다. ids가 RaggedTensor인 경우 embedding_lookup()의 ids.row_splits에 해당합니다. ND 비정형 텐서를 대기열에 넣을 때 마지막 차원만 비정형화될 수 있습니다. 행 분할은 1차원 밀집 텐서입니다. 비어 있으면 조밀한 텐서가 op에 전달된다고 가정합니다. int32와 int64가 모두 허용되며 내부적으로 int32로 변환됩니다. |
| 임베딩인덱스 | 임베딩 테이블에 대한 인덱스인 순위 1 Tensor의 목록입니다. int32와 int64가 모두 허용되며 내부적으로 int32로 변환됩니다. |
| 집계가중치 | 훈련 예시별 집계 가중치를 포함하는 순위 1 Tensor의 목록입니다. float32와 float64가 모두 허용되며 내부적으로 float32로 변환됩니다. |
| 모드 재정의 | TPUEmbeddingConfiguration에 지정된 모드를 재정의하는 문자열 입력입니다. 지원되는 값은 {'unspecified', 'inference', 'training', 'backward_pass_only'}입니다. 'unspecified'로 설정되면 TPUEmbeddingConfiguration에 설정된 모드가 사용되고, 그렇지 않으면 mode_override가 사용됩니다. |
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- EnqueueTPUEmbeddingArbitraryTensorBatch의 새로운 인스턴스
공개 정적 EnqueueTPUEmbeddingArbitraryTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
매개변수
| 장치서수 | 사용할 TPU 기기입니다. 0보다 크고 노드가 배치된 작업의 TPU 코어 수보다 작아야 합니다. |
|---|