Implement FourierLayer for higher dimensions#31

Closed

pzimbrod wants to merge 8 commits intomasterfrom

pzimbrod-FourierLayer-ND

Commits on Feb 11, 2022

Commits on Feb 15, 2022

💡 update FNO trainable function for multiple modes
pzimbrod
committed

Commits on Feb 16, 2022

Commits on Feb 17, 2022

get rid of unnecessary permutations
pzimbrod
committed

Commits on Feb 18, 2022

use TensorCore's boxdot in the linear path
pzimbrod
committed