I try to apt the merge(mode=‘avg’) of keras. What is the torch way of doing this?   I try to sum and divide by the count but it raises RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation
 def forward(self, inputs):
            outputs = []
            for x in inputs:
                x = self.conv_column(x)
                x = self.clf_layer(x)
                outputs.append(x) 
                # avg all outputs here !
            return outputs