Finds values and indices of the k
largest elements for the last dimension.
tf.raw_ops.TopK(
input, k, sorted=True, name=None
)
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.
If k
varies dynamically, use TopKV2
below.
Returns | |
---|---|
A tuple of Tensor objects (values, indices).
|
|
values
|
A Tensor . Has the same type as input .
|
indices
|
A Tensor of type int32 .
|