fftn_op

Structs


fwd(arg0: Array, dims: List[Int], norm: String) -> Array

Sets up the Array object for the FFT operation.


jvp(primals: List[Array], tangents: List[Array]) -> Array

Computes the Jacobian-vector product for the FFT function.


vjp(primals: List[Array], grad: Array, out: Array) -> List[Array]

Computes the vector-Jacobian product for the FFT function.


__call__(mut curr: Array, args: List[Array])

Executes the FFT operation inplace.


fftn(x: Array, dims: List[Int] = List(), norm: String = String("backward")) -> Array

Compute the n-dimensional FFT.

x: Array The input array.
dims: List[Int] (default: List()) The dimensions along which to compute the FFT.
norm: String (default: String("backward")) The normalization mode.