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

#include <nn_ops.h>

마지막 차원에 대해 가장 큰 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 의 인덱스입니다.

생성자와 소멸자

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)

공개 속성

indices
operation
values

공개 정적 함수

Sorted (bool x)

구조체

텐서플로우:: ops:: TopK:: 속성

TopK 에 대한 선택적 속성 설정자입니다.

공개 속성

지수 

::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
)