I want to  compute a coefficient for each channels, and then multiply the features with coefficient:
coefficient = self.bknet(feature)
feature[:,:,:] = feature[:,:,:]*coefficient[:,:,:]

where “feature” is output of some conv layers, but when I train the net, the error is:

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation

and I try to add

F.relu()

but it still not work.
thanks for your help!