텐서플로우:: ops:: TopK

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

요약

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

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

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

두 요소가 같을 경우, 인덱스가 낮은 요소가 먼저 나타납니다.

논거:

• 범위: Scope 객체
• 입력: 1차원 이상, 마지막 차원이 최소 k 이상.
• k: 0차원. 마지막 차원(행렬의 경우 각 행)을 따라 찾을 상위 요소의 개수입니다.

선택적 속성( 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
)