Closed pzimbrod closed 2 years ago
Following things had to be changed:
OMEinsum
. Looping over 3-dimensional arrays would suffice for a 1D Problem, but certainly not for higher-dimensional ones.OMEinsum
, gains were marginal at best anyway.false
.batchsize
argument out of the initialization function as it's no longer needed. This increases the flexibility of the layer considerably.CuFFT
poses and the requirements of Flux.DataLoader
at the same time.
Resolves #13