use truncate_modes as function name

Author: yuehhua

src/Transform/Transform.jl

@@ -18,5 +18,3 @@ abstract type AbstractTransform end
 
 include("fourier_transform.jl")
 include("chebyshev_transform.jl")
-
-const truncate_modes = low_pass

src/Transform/chebyshev_transform.jl

@@ -10,8 +10,8 @@ function transform(t::ChebyshevTransform{N}, 𝐱::AbstractArray) where {N}
     return FFTW.r2r(𝐱, FFTW.REDFT00, 1:N) # [size(x)..., in_chs, batch]
 end
 
-function low_pass(t::ChebyshevTransform, 𝐱̂::AbstractArray)
-    return view(𝐱̂, map(d->1:d, t.modes)..., :, :) # [ft.modes..., in_chs, batch]
+function truncate_modes(t::ChebyshevTransform, 𝐱̂::AbstractArray)
+    return view(𝐱̂, map(d->1:d, t.modes)..., :, :) # [t.modes..., in_chs, batch]
 end
 
 function inverse(t::ChebyshevTransform{N}, 𝐱̂::AbstractArray) where {N}

src/Transform/fourier_transform.jl

@@ -14,7 +14,7 @@ function low_pass(ft::FourierTransform, 𝐱_fft::AbstractArray)
     return view(𝐱_fft, map(d -> 1:d, ft.modes)..., :, :) # [ft.modes..., in_chs, batch]
 end
 
-truncate_modes(args...) = low_pass(args...)
+truncate_modes(ft::FourierTransform, 𝐱_fft::AbstractArray) = low_pass(ft, 𝐱_fft)
 
 function inverse(ft::FourierTransform, 𝐱_fft::AbstractArray)
     return real(ifft(𝐱_fft, 1:ndims(ft))) # [size(x_fft)..., out_chs, batch]