top_k¶
dragon.vm.tensorflow.math.
top_k
(
input,
k=1,
sorted=True,
name=None
)[source]¶Return the top-K largest elements along the last axis.
- Parameters:
- input (dragon.Tensor) – The input tensor.
- k (int, optional, default=1) – The number of top elements to select.
- sorted (bool, optional) – Whether to return in the sorted order.
- name (str, optional) – The operation name.
- Returns:
Sequence[dragon.vm.torch.Tensor] – The value and index tensor.