Mutli-dimensional input-grid for DeepONet

[dynamic-queries]

In https://github.com/SciML/NeuralOperators.jl/blob/master/src/DeepONet.jl#L122-L124, multi-dimensional input grid is handled using an Argument error. Is there a reason this was done this way?

Also, if it is a matter of contracting AbstractArray(s), we could look into OMEinsum.jl for its implemention.

Please advise.

[yuehhua]

multi-dimensional input grid is handled using an Argument error. Is there a reason this was done this way?

@Abhishek-1Bhatt Any ideas?

For contracting AbstractArray(s), this package uses Tullio.jl for it. Should work.

[dynamic-queries]

If this is still unresolved. I have a quick fix with OMEinsum.jl
Should I open a PR?

[ChrisRackauckas]

Could you try setting it up with Tullio? That should make it easier for making it support GPUs.

[yuehhua]

It is welcome to support multi-dimensional input for DeepONet. However, it will be nice to use Tullio to make deps simple.