diff options
Diffstat (limited to 'tensorflow/g3doc/api_docs/python/functions_and_classes/shard5/tf.nn.top_k.md')
-rw-r--r-- | tensorflow/g3doc/api_docs/python/functions_and_classes/shard5/tf.nn.top_k.md | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/tensorflow/g3doc/api_docs/python/functions_and_classes/shard5/tf.nn.top_k.md b/tensorflow/g3doc/api_docs/python/functions_and_classes/shard5/tf.nn.top_k.md new file mode 100644 index 0000000000..819c0ad068 --- /dev/null +++ b/tensorflow/g3doc/api_docs/python/functions_and_classes/shard5/tf.nn.top_k.md @@ -0,0 +1,31 @@ +### `tf.nn.top_k(input, k=1, sorted=True, name=None)` {#top_k} + +Finds values and indices of the `k` largest entries for the last dimension. + +If the input is a vector (rank-1), finds the `k` largest entries in the vector +and outputs their values and indices as vectors. Thus `values[j]` is the +`j`-th largest entry in `input`, and its index is `indices[j]`. + +For matrices (resp. higher rank input), computes the top `k` entries in each +row (resp. vector along the last dimension). Thus, + + values.shape = indices.shape = input.shape[:-1] + [k] + +If two elements are equal, the lower-index element appears first. + +##### Args: + + +* <b>`input`</b>: 1-D or higher `Tensor` with last dimension at least `k`. +* <b>`k`</b>: 0-D `int32` `Tensor`. Number of top elements to look for along the last + dimension (along each row for matrices). +* <b>`sorted`</b>: If true the resulting `k` elements will be sorted by the values in + descending order. +* <b>`name`</b>: Optional name for the operation. + +##### Returns: + + +* <b>`values`</b>: The `k` largest elements along each last dimensional slice. +* <b>`indices`</b>: The indices of `values` within the last dimension of `input`. + |