class_docstr: Return the highest scoring action.
class_docstr_tok: ["Return","the","highest","scoring","action","."]
code: def _get_greedy_indices(self, scores: torch.Tensor) -> torch.Tensor:
      assert (
          len(scores.shape) == 2
      ), f"scores shape is {scores.shape}, not (batchsize, num_actions)"
      return scores.argmax(dim=1)
  
imports: import reagent.core.types
  import torch
  import torch.nn.functional
  from reagent.gym.types import Sampler
  from reagent.models.dqn import INVALID_ACTION_CONSTANT
label_desc: Use image node_id 1 for calling the GreedyActionSampler obj's underlying member method code with example usage: obj._get_greedy_indices(scores) and returns: scores