Add option in tf.gradients() to return zero tensors for unconnected gradients.
tf.gradients currently returns [NONE] when the gradient of unconnected variables is required. This backwards compatable change adds in the option to have zero tensors returned that match the dimensions of the input tensor. PiperOrigin-RevId: 215725488
Loading
Please sign in to comment