Could you post a minimal, executable code snippet using random input tensors, which would show the unexpected difference between the functional and module approach, please?
You can add code snippets by wrapping them into three backticks ```, which makes debugging easier