def sqrt(tensor, *args, **kwargs): if isinstance(tensor, nd.NDArray): return nd.sqrt(tensor, *args, **kwargs) else: return scalar_sqrt(tensor)
def sqrt(tensor, *args, **kwargs): if torch.is_tensor(tensor): return torch.sqrt(tensor, *args, **kwargs) else: return scalar_sqrt(tensor)
def get_sig_norm(self, sctx, eps_app_eng): sig_eng, D_mtx = self.get_corr_pred(sctx, eps_app_eng, 0, 0, 0) return array([ scalar_sqrt(sig_eng[0] ** 2 + sig_eng[1] ** 2) ])
def get_sig_norm(self, sctx, eps_app_eng): sig_eng, D_mtx = self.get_corr_pred(sctx, eps_app_eng, 0, 0, 0) return array([scalar_sqrt(sig_eng[0] ** 2 + sig_eng[1] ** 2)])