code: def sqrtm(x):
      np_sqrtm = _np.vectorize(
          _scipy.linalg.sqrtm, signature="(n,m)->(n,m)"
      )(x)
      if np_sqrtm.dtype.kind == "c":
          np_sqrtm = np_sqrtm.astype(
              f"complex{int(np_sqrtm.dtype.name[7:]) // 2}"
          )
  
      return _torch.from_numpy(np_sqrtm)
  
file_path: geomstats/geomstats/_backend/pytorch/linalg.py
imports: import numpy
  import scipy
  import torch
  from .._backend_config import np_atol
  from ..numpy import linalg
  from ._dtype import _cast_out_to_input_dtype
label_desc: Use image node_id 1 for calling a global function with example usage: sqrtm(x) and returns: _torch