텐서플로우:: 작전:: 인탑케이
#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
xclasses
텐서. - 타겟: 클래스 ID의
batch_size
벡터입니다. - k: 계산 정밀도를 위해 살펴볼 상위 요소 수입니다.
보고:
생성자와 소멸자 | |
---|---|
InTopK (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, int64 k) |
공개 속성 | |
---|---|
operation | |
precision |
공공 기능 | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
공개 속성
작업
Operation operation
정도
::tensorflow::Output precision
공공 기능
인탑케이
InTopK( const ::tensorflow::Scope & scope, ::tensorflow::Input predictions, ::tensorflow::Input targets, int64 k )
마디
::tensorflow::Node * node() const
연산자::텐서플로우::입력
operator::tensorflow::Input() const
연산자::텐서플로우::출력
operator::tensorflow::Output() const