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

Commit 9917824

Browse files
committed
Refactor docstring
1 parent edfa983 commit 9917824

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/operator_kernel.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@ end
3333
3434
* `init`: Initial function to initialize parameters.
3535
* `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)`.
36+
data in the order of `(ch, x_1, ... , x_d , batch)`,
37+
otherwise the order is `(x_1, ... , x_d, ch, batch)`.
3738
* `T`: Data type of parameters.
3839
3940
## Example
@@ -133,7 +134,8 @@ end
133134
## Keyword Arguments
134135
135136
* `permuted`: Whether the dim is permuted. If `permuted=true`, layer accepts
136-
data in the order of `(ch, ..., batch)`, otherwise the order is `(..., ch, batch)`.
137+
data in the order of `(ch, x_1, ... , x_d , batch)`,
138+
otherwise the order is `(x_1, ... , x_d, ch, batch)`.
137139
138140
## Example
139141

0 commit comments

Comments
 (0)