How to use multiple inputs in `NeuralODE` object?

oh sorry I did not make it clear. By saying multiple inputs, I meant “each data point has multiple inputs”, an example is this neural network structure: https://machinelearningmastery.com/wp-content/uploads/2017/08/multiple_inputs.png, where we have two branches of inputs and one output. NeuralODE does not seem to be able to accept more than one branch of inputs, so I am wondering whether there is a way to do so.

UPDATE: I figure out how to pass multiple inputs as a tuple, but it looks like trainable parameters are not correctly captured, as discussed here: How to make parameters of function within Flux.Chain trainable?, if you can take a look at this question, that would be great, thanks!

1 Like