RecvTPUEmbeddingActivations

공개 최종 클래스 RecvTPUEmbeddingActivations

TPU에서 임베딩 활성화를 수신하는 작업입니다.

TPU 시스템은 TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch에 대한 인수로 지정된 임베딩 조회 및 집계를 수행합니다. 이러한 집계 결과는 RecvTPUEmbeddingActivations 작업의 출력으로 Tensorflow 그래프에 표시됩니다. 이 작업은 모델에 지정된 테이블당 하나의 활성화 Tensor를 포함하는 목록을 반환합니다. TPU 그래프에는 RecvTPUEmbeddingActivations 작업이 최대 1개 있을 수 있습니다.

공개 방법

정적 RecvTPUEmbeddingActivations
생성 ( 범위 범위, Long numOutputs, 문자열 구성)
새로운 RecvTPUEmbeddingActivations 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
반복자< 피연산자 <부동 소수점>>
목록< 출력 <플로트>>
출력 ()
모델의 임베딩 테이블당 하나의 Tensor를 포함하는 임베딩 활성화의 TensorList입니다.

상속된 메서드

공개 방법

공개 정적 RecvTPUEmbeddingActivations 생성 ( 범위 범위, Long numOutputs, 문자열 구성)

새로운 RecvTPUEmbeddingActivations 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
출력 수 출력 활성화 텐서의 수는 모델의 임베딩 테이블 수와 같습니다.
구성 직렬화된 TPUEmbeddingConfiguration proto.
보고
  • RecvTPUEmbeddingActivations의 새 인스턴스

공개 반복자< 피연산자 <Float>> 반복자 ()

공개 목록< 출력 <Float>> 출력 ()

모델의 임베딩 테이블당 하나의 Tensor를 포함하는 임베딩 활성화의 TensorList입니다.