code: def _randn(prototype, shape, rng_key=None):
      assert isinstance(shape, tuple)
      return torch.randn(
          shape, dtype=prototype.dtype, device=prototype.device
      )
  
imports: import numbers
  import typing
  import torch
  import funsor.ops
label_desc: Use image node_id 40 for calling a global function with example usage: _randn(prototype, shape, rng_key) and returns: torch