Skip to content
This repository was archived by the owner on Sep 28, 2024. It is now read-only.

Commit 8254713

Browse files
committed
add permuted args
1 parent 77c3da0 commit 8254713

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/fourier.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,8 @@ end
3232
* `ch`: Input and output channel size, e.g. `64=>64`.
3333
* `modes`: The Fourier modes to be preserved.
3434
* `σ`: Activation function.
35+
* `permuted`: Whether the dim is permuted. If `permuted=true`, layer accepts
36+
data in the order of `(..., ch, batch)`, otherwise the order is `(ch, ..., batch)`.
3537
3638
## Example
3739
@@ -112,6 +114,8 @@ end
112114
* `ch`: Input and output channel size for spectral convolution, e.g. `64=>64`.
113115
* `modes`: The Fourier modes to be preserved for spectral convolution.
114116
* `σ`: Activation function.
117+
* `permuted`: Whether the dim is permuted. If `permuted=true`, layer accepts
118+
data in the order of `(..., ch, batch)`, otherwise the order is `(ch, ..., batch)`.
115119
116120
## Example
117121

0 commit comments

Comments
 (0)