ComputeDedupDataTupleMask

공개 최종 클래스 ComputeDedupDataTupleMask

연산은 임베딩 코어에서 중복 제거 데이터의 튜플 마스크를 계산합니다.

임베딩 코어에서 수신되는 중복 제거 데이터는 유형=DT_VARIANT인 Tensor입니다. 텐서 자체는 요소가 순위 1 텐서인 XLA 중첩 튜플입니다. 이 작업은 이러한 요소의 유형과 길이를 나타냅니다.

공개 방법

출력 <정수>
출력 ()
텐서의 기호 핸들을 반환합니다.
정적 ComputeDedupDataTupleMask
생성 ( 범위 범위, 문자열 구성)
새로운 ComputeDedupDataTupleMask 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <정수>
출력모양 ()
2-D int 텐서는 `XlaRecvTPUEmbeddingDeduplicationData`에 의해 생성된 중복 제거 데이터 튜플의 마스크를 나타냅니다.

상속된 메서드

공개 방법

공개 출력 <Integer> asOutput ()

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

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

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

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

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

공개 출력 <Integer> outputShape ()

2-D int 텐서는 `XlaRecvTPUEmbeddingDeduplicationData`에 의해 생성된 중복 제거 데이터 튜플의 마스크를 나타냅니다. 튜플에는 여러 정수 및 부동 소수점 유형의 1차원 텐서 튜플 요소가 있습니다. 이 output_shape 2차원 텐서의 첫 번째 차원은 튜플 요소의 텐서 유형입니다. '0'은 정수 텐서를 나타내고, '1'은 부동 텐서를 나타냅니다. `output_shape`의 두 번째 차원은 각 튜플 요소의 길이를 제공합니다.