code: def _batch_fake_probs(indexes, count):
      p_seq = torch.zeros(indexes.shape + (count,))
  
      for batch_idx in range(len(indexes)):
          for item_idx in range(indexes.size(1)):
              p_seq[batch_idx, item_idx, :] = _fake_probs(
                  indexes[batch_idx, item_idx], count
              )
      return p_seq
  
imports: import torch
  from torch.nn import functional
label_desc: Use image node_id 2 for calling a global function with example usage: _batch_fake_probs(indexes, count) and returns: p_seq