This repository was archived by the owner on Sep 28, 2024. It is now read-only.
Repository navigation
This repository was archived by the owner on Sep 28, 2024. It is now read-only.
Mutli-dimensional input-grid for DeepONet #73
Description
Activity
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.
Reacted by Rahul ManavalanIf this is still unresolved. I have a quick fix with OMEinsum.jl
Should I open a PR?Could you try setting it up with Tullio? That should make it easier for making it support GPUs.
Reacted by Yueh-Hua Tu and Rahul ManavalanIt is welcome to support multi-dimensional input for DeepONet. However, it will be nice to use Tullio to make deps simple.
Reacted by Rahul Manavalan
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.