From 4e642358c6e493240a1fba7dfe4848c3767f445a Mon Sep 17 00:00:00 2001 From: JingYu Ning Date: Mon, 22 Nov 2021 20:03:11 +0800 Subject: [PATCH] Update index.md --- docs/src/index.md | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/docs/src/index.md b/docs/src/index.md index d8c9f8c9..4154ca3c 100644 --- a/docs/src/index.md +++ b/docs/src/index.md @@ -46,14 +46,15 @@ pkg> add NeuralOperators ```julia model = Chain( - # project finite-dimensional data to infinite-dimensional space + # lift (d + 1)-dimensional vector field to n-dimensional vector field + # here, d == 1 and n == 64 Dense(2, 64), - # operator projects data between infinite-dimensional spaces + # map each hidden representation to the next by integral kernel operator FourierOperator(64=>64, (16, ), gelu), FourierOperator(64=>64, (16, ), gelu), FourierOperator(64=>64, (16, ), gelu), FourierOperator(64=>64, (16, )), - # project infinite-dimensional function to finite-dimensional space + # project back to the scalar field of interest space Dense(64, 128, gelu), Dense(128, 1), flatten