Hi all. I have 3 different convolution blocks each with channel number 64. I would like to make an element wise summation with trainable weights for each of the convolution blocks, i.e.
let conv_1 , conv_2 and conv_3 be the convolution blocks.
conv_final = lambda_1 * conv_1 + lambda_2* conv_2 + lambda_3* conv_3 (+ here means element wise summation)
I want to train these lambda’s while i train my whole CNN using the loss function. Can somebody please help me in implementing this?