tensorflow :: ops :: TopK

마지막 차원에 대해 k 가장 큰 요소의 값과 인덱스를 찾습니다.

요약

입력 값이 벡터 (순위 -1) 인 경우 벡터에서 k 가장 큰 항목을 찾고 해당 값과 인덱스를 벡터로 출력합니다. 따라서 values[j] 는 input 에서 j 번째로 큰 항목이고 인덱스는 indices[j] 입니다.

행렬 (상위 순위 입력)의 경우 각 행의 상위 k 개 항목 (각각 마지막 차원을 따라 벡터)을 계산합니다. 따라서

values.shape = indices.shape = input.shape[:-1] + [k]

두 요소가 같으면 하위 인덱스 요소가 먼저 나타납니다.

인수 :

• 범위 : 범위 개체
• 입력 : 마지막 차원이 k 이상인 1 차원 이상.
• k : 0-D. 마지막 차원에서 찾을 상위 요소의 수 (행렬의 각 행).

선택적 속성 ( Attrs 참조) :

• sorted : true이면 결과 k 요소가 값을 기준으로 내림차순으로 정렬됩니다.

보고:

• Output 값 : 각 마지막 차원 슬라이스를 따라 가장 큰 k 요소.
• Output 색인 : input 의 마지막 차원 내 values 의 색인입니다.

공용 속성

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Attrs Sorted(
  bool x
)