I am training a neural networks with three different output prediction. For computing the loss of one output I need one of the input that is passed into the network. I am no