|
if False: |
|
print("Ws") |
|
E, Ex, Ey = torch.split(E_out,1,dim=1) |
|
El, Elx, Ely = torch.split(El_out,1,dim=1) |
|
Ell, Ellx, Elly = torch.split(Ell_out,1,dim=1) |
|
|
|
H = hat_H(E,Ex,Ey,El,None,None,Ell,None,None) |
|
S = torch.log(hat_S(E,Ex,Ey,El,None,None,Ell,None,None)+eps) |
|
Ww = torch.atan(hat_Ww(E,Ex,Ey,El,None,None,Ell,None,None) + \ |
|
hat_Wlw2(E,Ex,Ey,El,Elx,Ely,Ell,Ellx,Elly) + \ |
|
hat_Wllw2(E,Ex,Ey,El,Elx,Ely,Ell,Ellx,Elly)) |
|
else: |
|
# print("W1") |
|
E, Ex, Ey = torch.split(E_out,1,dim=1) |
|
El = torch.split(El_out,1,dim=1)[0] |
|
Ell = torch.split(Ell_out,1,dim=1)[0] |
|
|
|
H = hat_H(E,Ex,Ey,El,None,None,Ell,None,None) |
|
S = torch.log(hat_S(E,Ex,Ey,El,None,None,Ell,None,None)+eps) |
|
Ww = torch.atan(hat_Ww(E,Ex,Ey,El,None,None,Ell,None,None)) |
Why there is if False? Do they have any special meaning?
QuadPrior/ciconv2d.py
Lines 134 to 153 in cbdf02f