code: def test_select_output_token_sample(self):
      result = sample_output_token(
          torch.tensor([[0.0, 0.5, 1.0]]), True, 1, 1, 1.0
      )
      assert result == torch.tensor(2)
  
file_path: ecco/tests/lm_test.py
imports: from ecco.lm import LM, _one_hot, sample_output_token, activations_dict_to_array
  import ecco
  import torch
  import numpy
  from transformers import PreTrainedModel
label_desc: Use image node_id 3 for calling the TestLM obj's underlying member method code with example usage: obj.test_select_output_token_sample() without return types