XlaRecvTPUEmbeddingDeduplicationData

공개 최종 클래스 XlaRecvTPUEmbeddingDeduplicationData

임베딩 코어로부터 중복 제거 데이터(인덱스 및 가중치)를 수신합니다.

중복 제거 데이터는 유형=DT_VARIANT인 Tensor입니다. 텐서 자체는 N 요소를 포함하는 XLA 중첩 튜플입니다. 여기서 N은 TPU 칩당 텐서 코어에 대한 임베딩 수의 비율입니다. 중첩된 튜플의 각 요소는 순위 1 텐서의 튜플입니다. 각 텐서는 TensorCore에 임베딩 조회를 위한 인덱스(DT_UINT32) 또는 임베딩 조회 작업의 출력에 적용할 가중치(DT_FLOAT)를 포함합니다.

공개 방법

<객체> 출력
출력 ()
텐서의 기호 핸들을 반환합니다.
정적 XlaRecvTPUEmbeddingDeduplicationData
생성 ( 범위 범위, 문자열 구성)
새로운 XlaRecvTPUEmbeddingDeduplicationData 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <?>
출력 ()

상속된 메서드

공개 방법

공개 출력 <Object> asOutput ()

텐서의 기호 핸들을 반환합니다.

TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.

공개 정적 XlaRecvTPUEmbeddingDeduplicationData 생성 ( 범위 범위, 문자열 구성)

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

매개변수
범위 현재 범위
구성 직렬화된 TPUEmbeddingConfiguration proto.
보고
  • XlaRecvTPUEmbeddingDeduplicationData의 새 인스턴스

공개 출력 <?> 출력 ()