code: def _torch_empty(
      batch_size: int, n_gates: int, device: torch.device
  ) -> Tensor:
      return torch.empty(batch_size, n_gates, device=device)
  
file_path: captum/captum/module/binary_concrete_stochastic_gates.py
imports: import math
  from typing import Optional
  import torch
  from captum.module.stochastic_gates_base import StochasticGatesBase
  from torch import nn, Tensor
label_desc: Use image node_id 1 for calling a global function with example usage: _torch_empty(batch_size, n_gates, device) and returns: torch