code: def bsh_decoder_scatter(hidden_states, part_hidden_states, rand_list):
      for k in range(hidden_states.size(0)):
          hidden_states[k, rand_list[k], :] = part_hidden_states[
              k, :, :
          ]
      return hidden_states
  
label_desc: Use image node_id 2 for calling a global function with example usage: bsh_decoder_scatter(hidden_states, part_hidden_states, rand_list) and returns: hidden_states