텐서플로우:: 작전:: 탑케이

마지막 차원에 대해 가장 큰 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
)