Commit b8af54d3 authored by Tamara Norman's avatar Tamara Norman Committed by TensorFlower Gardener
Browse files

Add option to tf.GradientTape gradient to return zero tensor for unconnected tensors. 

Added unconnected gradients argument to GradientTape.gradient linked to the same argument as the tf.gradients module. This allows the option to return a zero tensor instead of None when taking the gradient of unconnected tensors

PiperOrigin-RevId: 218186755
parent 5bc359be
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment