тензорный поток:: опс:: ИнТопК
#include <nn_ops.h>
Сообщает, входят ли цели в топ K
прогнозов.
Краткое содержание
В результате выводится массив bool batch_size
, запись out[i]
имеет значение true
, если прогноз для целевого класса входит в число k
лучших прогнозов среди всех прогнозов, например i
. Обратите внимание, что поведение InTopK
отличается от поведения TopK
в обработке связей; если несколько классов имеют одинаковое значение прогноза и находятся на границе верхнего k
, все эти классы считаются находящимися в верхнем k
.
Более формально, пусть
\(predictions_i\) быть прогнозами для всех классов, например i
, \(targets_i\) быть целевым классом, например i
, \(out_i\) быть результатом, например, i
,
Аргументы:
- область: объект области.
- прогнозы: тензор
classes
batch_size
x. - цели: вектор
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
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const