code: def _fake_probs(idx, count):
      result = torch.zeros(count)
      result[idx] = 2.0
      return F.softmax(result, dim=-1)
  
imports: import torch
  from torch.nn import functional
label_desc: Use image node_id 1 for calling a global function with example usage: _fake_probs(idx, count) and returns: F