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.