テンソルフロー::作戦::イントップK

#include <nn_ops.h>

ターゲットが上位K予測に含まれるかどうかを示します。

まとめ

これは、 batch_sizeブール配列を出力します。ターゲット クラスの予測がすべての予測の中で上位kの予測に含まれる場合、エントリout[i] trueになります (例: i 。 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として計算された精度。

コンストラクターとデストラクター

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

公共機能

イントップK 

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

ノード

::tensorflow::Node * node() const

演算子::tensorflow::入力

 operator::tensorflow::Input() const

演算子::tensorflow::出力

 operator::tensorflow::Output() const