Commit 8b91877f authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Fix SVD gradient bug on batches of matrices 

Fix a newly introduced bug in dealing with a gradient of SVD applied to a batch of matrices.
s is a batch of vectors, not matrices, so we should use shape[:-1] to get the batch shape.

PiperOrigin-RevId: 236134183
parent cd233817
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