code: def _torch_sqrt(x, eps=1e-12):
      """
      A convenient function to avoid the NaN gradient issue of :func:`torch.sqrt`
      at 0.
      """
      # Ref: https://github.com/pytorch/pytorch/issues/2421
      return (x + eps).sqrt()
  
imports: import torch
  from torch.distributions import constraints
  from pyro.contrib.gp.kernels.kernel import Kernel
  from pyro.nn.module import PyroParam
label_desc: Use image node_id 1 for calling a global function with example usage: _torch_sqrt(x, eps) and returns: unknown