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.