텐서플로우:: 작전:: 인탑KV2

#include <nn_ops.h>

대상이 상위 K 개 예측에 있는지 여부를 나타냅니다.

요약

이는 batch_size 부울 배열을 출력하며, 대상 클래스에 대한 예측이 예를 들어 i 와 같이 모든 예측 중 상위 k 개 예측 중 하나인 경우 항목 out[i] 는 true 입니다. InTopK 의 동작은 동점 처리에 있어서 TopK 작업과 다릅니다. 여러 클래스가 동일한 예측 값을 갖고 상위 k 경계에 걸쳐 있는 경우 해당 클래스는 모두 상위 k 에 있는 것으로 간주됩니다.

좀 더 공식적으로 보자면

\(predictions_i\) 모든 클래스에 대한 예측입니다(예: i , \(targets_i\) 는 대상 클래스(예: i , \(out_i\) 은 예를 들어 i 에 대한 출력입니다.

$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$

인수:

  • 범위: 범위 개체
  • 예측: batch_size x classes 텐서.
  • 타겟: 클래스 ID의 batch_size 벡터입니다.
  • k: 계산 정밀도를 위해 살펴볼 상위 요소 수입니다.

보고:

  • Output : k 에서 bool Tensor 로 정밀도를 계산했습니다.

생성자와 소멸자

InTopKV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, :: tensorflow::Input k)

공개 속성

operation
precision

공공 기능

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

공개 속성

작업

Operation operation

정도

::tensorflow::Output precision

공공 기능

인탑KV2

 InTopKV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input predictions,
  ::tensorflow::Input targets,
  ::tensorflow::Input k
)

마디

::tensorflow::Node * node() const

연산자::텐서플로우::입력

 operator::tensorflow::Input() const

연산자::텐서플로우::출력

 operator::tensorflow::Output() const