공개 최종 수업 ApproxTopK
입력 피연산자의 최소/최대 k 값과 해당 인덱스를 대략적인 방식으로 반환합니다.
알고리즘 세부정보는 https://arxiv.org/abs/2206.14286을 참조하세요. 이 작업은 현재 TPU에만 최적화되어 있습니다.
중첩 클래스
수업 | ApproxTopK.옵션 | ApproxTopK 의 선택적 속성 |
공개 방법
정적 ApproxTopK.Options | AggregateToTopk (부울 집계ToTopk) |
static <T 확장 숫자> ApproxTopK <T> | |
출력 <정수> | 인덱스 () '입력' 피연산자의 'reduction_dimension'에 따른 '값'의 인덱스입니다. |
정적 ApproxTopK.Options | isMaxK (부울 isMaxK) |
정적 ApproxTopK.Options | RememberTarget (플로트 리콜 타겟) |
정적 ApproxTopK.Options | 감소차원 (긴 감소차원) |
정적 ApproxTopK.Options | 감소InputSizeOverride (긴 감소InputSizeOverride) |
출력 <T> | 값 () 'input' 피연산자의 'reduction_dimension'에 따른 최소/최대 k 값입니다. |
상속된 메서드
공개 방법
공개 정적 ApproxTopK.Options 집계ToTopk (부울 집계ToTopk)
매개변수
AggregateToTopk | true인 경우 대략적인 결과를 top-k로 집계합니다. false인 경우 대략적인 결과를 반환합니다. 대략적인 결과의 수는 구현에 따라 정의되며 지정된 'k'보다 큽니다. |
---|
공개 정적 ApproxTopK <T> 생성 ( 범위 범위, 피연산자 <T> 입력, Long k, 옵션... 옵션)
새로운 ApproxTopK 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|---|
입력 | 검색할 배열입니다. 부동 유형의 1D 이상이어야 합니다. |
케이 | 최소/최대-k 수를 지정합니다. |
옵션 | 선택적 속성 값을 전달합니다. |
보고
- ApproxTopK의 새 인스턴스
공개 정적 ApproxTopK.Options 감소InputSizeOverride (긴 감소InputSizeOverride)
매개변수
감소입력크기재정의 | 양수 값으로 설정하면 리콜 평가를 위해 `input[reduction_dim]`에 의해 결정된 크기를 재정의합니다. 이 옵션은 주어진 '입력'이 SPMD 또는 분산 파이프라인의 전체 계산의 하위 집합일 뿐이고 실제 입력 크기가 '입력' 모양으로 지연될 수 없는 경우에 유용합니다. |
---|
공개 출력 <T> 값 ()
'input' 피연산자의 'reduction_dimension'에 따른 최소/최대 k 값입니다. 차원은 `reduction_dimension`을 제외하고 `input` 피연산자와 동일합니다. `aggregate_to_topk`가 true인 경우 축소 차원은 `k`입니다. 그렇지 않으면 크기가 구현에 따라 정의되는 'k'보다 큽니다.