@@ -33,7 +33,7 @@ which is useful when imposing scale-dependent regularization
33
33
"""
34
34
function Aodwt (
35
35
dims:: Dims ;
36
- T:: Type{<:Complex{<:AbstractFloat}} = ComplexF32 ,
36
+ T:: Type{<:Union{AbstractFloat, Complex{<:AbstractFloat}}} = Float32 ,
37
37
level:: Int = 3 ,
38
38
wt = wavelet (WT. haar),
39
39
operator:: Bool = true , # !
@@ -52,7 +52,7 @@ function Aodwt(
52
52
mfun = (A, x) -> A * x
53
53
return mfun, LinearMapAA (forw!, back!,
54
54
(prod (dims), prod (dims)) ;
55
- prop = (wt= wt , level= lev), T = T,
55
+ prop = (wt, level= lev), T,
56
56
operator = true , idim= dims, odim= dims,
57
57
)
58
58
else
@@ -63,7 +63,7 @@ function Aodwt(
63
63
# x -> vec(dwt(reshape(x, dims), wt, level)),
64
64
# y -> vec(idwt(reshape(y, dims), wt, level)),
65
65
(prod (dims), prod (dims)) ;
66
- prop = (wt= wt , level= lev), T = T,
66
+ prop = (wt, level= lev), T,
67
67
)
68
68
end
69
69
end
@@ -72,7 +72,7 @@ function Aodwt(
72
72
73
73
scales = zeros (dims)
74
74
for il in 1 : level
75
- _,Al = mfunA (il)
75
+ _, Al = mfunA (il)
76
76
tmp = mfun (Al, ones (dims)) .== 0
77
77
scales += il * (tmp .& (scales .== 0 ))
78
78
end
0 commit comments