コード例 #1
0
def sqrt(tensor, *args, **kwargs):
    if isinstance(tensor, nd.NDArray):
        return nd.sqrt(tensor, *args, **kwargs)
    else:
        return scalar_sqrt(tensor)
コード例 #2
0
def sqrt(tensor, *args, **kwargs):
    if torch.is_tensor(tensor):
        return torch.sqrt(tensor, *args, **kwargs)
    else:
        return scalar_sqrt(tensor)
コード例 #3
0
 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) ])
コード例 #4
0
 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)])