class_docstr: Implements TopKOp on gpu
  
  Currently the output seem sorted, but we do not test it. So as on
  the CPU, we only support sorted=False for now.
class_docstr_tok: ["Implements","TopKOp","on","gpu","Currently","the","output","seem","sorted",",","but","we","do","not","test","it",".","So","as","on","the","CPU",",","we","only","support","sorted=False","for","now","."]
code: def __init__(
      self,
      axis=-1,
      sorted=True,
      idx_dtype="int64",
      return_values=True,
      return_indices=True,
  ):
      if sorted:
          raise NotImplementedError(
              "GpuTopK currently is not sure to give sorted output even if they look sorted.."
          )
      GpuKernelBase.__init__(self)
      TopKOp.__init__(
          self,
          axis=axis,
          sorted=sorted,
          idx_dtype=idx_dtype,
          return_values=return_values,
          return_indices=return_indices,
      )
  
imports: from __future__ import absolute_import, print_function, division
  import os
  from string import Template
  import numpy
  import theano
  from theano import Apply
  from theano.tensor import as_tensor_variable
  from theano.tensor.sort import TopKOp
  from .basic_ops import GpuKernelBase, Kernel, infer_context_name, as_gpuarray_variable, gpuarray_helper_inc_dir
  from .opt import register_opt, op_lifter, register_opt2
  from .type import GpuArrayType
label_desc: Use image node_id 1 to create a new GpuTopKOp object from inherited base classes: GpuKernelBase, TopKOp with example: obj = GpuTopKOp(axis, sorted, idx_dtype, return_values, return_indices)