How to handle multiple inputs in flax, is there any example as referece?

[long1216]

Is there any way to handle two inputs in flax(like the keras example showing above). Would there be any example showing how to do this? Thank you so much.

[marcvanzee]

The seq2seq model has two inputs: https://github.com/google/flax/blob/main/examples/seq2seq/train.py#L262

You simple have two arguments to the __call__ function of your Module.

Then when you want to run the forward pass you simply provide both arguments to the apply function of the Module.