Browse Source

Add tf.nn.in_top_k

Add tf.nn.in_top_k
tags/v0.12
Eli Belash GitHub 6 years ago
parent
commit
9d52c9eb0b
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      src/TensorFlowNET.Core/APIs/tf.nn.cs

+ 3
- 0
src/TensorFlowNET.Core/APIs/tf.nn.cs View File

@@ -131,6 +131,9 @@ namespace Tensorflow
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);



Loading…
Cancel
Save