공개 최종 클래스 RecvTPUEmbeddingActivations
TPU에서 임베딩 활성화를 수신하는 작업입니다.
TPU 시스템은 TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch에 대한 인수로 지정된 임베딩 조회 및 집계를 수행합니다. 이러한 집계 결과는 RecvTPUEmbeddingActivations 작업의 출력으로 Tensorflow 그래프에 표시됩니다. 이 작업은 모델에 지정된 테이블당 하나의 활성화 Tensor를 포함하는 목록을 반환합니다. TPU 그래프에는 RecvTPUEmbeddingActivations 작업이 최대 1개 있을 수 있습니다.
상수
끈 | OP_NAME | TensorFlow 코어 엔진에서 알려진 이 작업의 이름 |
공개 방법
정적 RecvTPUEmbeddingActivations | |
반복자< 피연산자 < TFloat32 >> | 반복자 () |
목록< 출력 < TFloat32 >> | 출력 () 모델의 임베딩 테이블당 하나의 Tensor를 포함하는 임베딩 활성화의 TensorList입니다. |
상속된 메서드
상수
공개 정적 최종 문자열 OP_NAME
TensorFlow 코어 엔진에서 알려진 이 작업의 이름
상수 값: "RecvTPUEmbeddingActivations"
공개 방법
공개 정적 RecvTPUEmbeddingActivations 생성 ( 범위 범위, Long numOutputs, 문자열 구성)
새로운 RecvTPUEmbeddingActivations 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
출력 수 | 출력 활성화 텐서의 수는 모델의 임베딩 테이블 수와 같습니다. |
구성 | 직렬화된 TPUEmbeddingConfiguration proto. |
보고
- RecvTPUEmbeddingActivations의 새 인스턴스