날짜를 저장하십시오! Google I / O가 5 월 18 일부터 20 일까지 반환됩니다. 지금 등록
이 페이지는 Cloud Translation API를 통해 번역되었습니다.
Switch to English

tensorflow :: ops :: TopK

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

구조체

tensorflow :: ops :: TopK :: Attrs

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

공용 속성

지수

::tensorflow::Output indices

조작

Operation operation

가치

::tensorflow::Output values

공공 기능

TopK

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

TopK

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

공개 정적 함수

정렬 됨

Attrs Sorted(
  bool x
)