Hi,
I’m able to circumvent this issue by sending a pseudo input that uses all the neural networks, and then I multiply the output of this input by zero. Since this input’s output doesn’t depend upon value of input, its gradient is 0.