Commit a1a370cb authored by Shanqing Cai's avatar Shanqing Cai Committed by TensorFlower Gardener
Browse files

Improve doc string and error checking in Permute Layer 

Check the validity of the permutation dims in the constructor of
tf.keras.layers.Permute. This improves the error messages
on invalid dims.

PiperOrigin-RevId: 207712499
parent aed8f42b
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