Add tf.nn.in_top_k

BrendanMulcahy · BrendanMulcahy · commit 098f597e32ae · 2019-08-18T13:44:06.000-04:00
diff --git a/src/TensorFlowNET.Core/APIs/tf.nn.cs b/src/TensorFlowNET.Core/APIs/tf.nn.cs
@@ -131,6 +131,8 @@ public static Tensor[] fused_batch_norm(Tensor x,
             public static Tensor max_pool(Tensor value, int[] ksize, int[] strides, string padding, string data_format = "NHWC", string name = null) 
                 => nn_ops.max_pool(value, ksize, strides, padding, data_format: data_format, name: name);
 
+            public static Tensor in_top_k(Tensor predictions, Tensor targets, int k, string name = "InTopK") => gen_ops.in_top_k(predictions, targets, k, name);
+
             public static Tensor[] top_k(Tensor input, int k = 1, bool sorted = true, string name = null)
                 => gen_nn_ops.top_kv2(input, k: k, sorted: sorted, name: name);
 
