Create two custom layers in tensorflow that add two inputs and multiply two inputs respectively [or any other mathematical operations of your choice(linear or nonlinear)] Add layer names and other things that are present in other common layers in tensorflow. Combine these two layers in a third custom layer. Concatenate them or multiply them etc. Create a model using these and observe how batch inference works