how to use bnn for custom function?

Thank you very much for this very descriptive and detailed implementation of BNNs. I am trying to implement BNN for a custom function which takes multiple inputs and produces one output. How can I define noise in such a case?