Commit c36fdd1c authored by Yong Tang's avatar Yong Tang
Browse files

Add unspecified dimensions (-1) support for noise_shape with tf.nn.dropout



This fix tries to address the issue raised in 16034 where
it was not possible to have unspecified dimensions
for `noise_shape` with `tf.nn.dropout`.

This fix adds the support so that it is possible to specify
`noise_shape = [-1, 1, 1, -1]` instead of `noise_shape = [k, 1, 1, n]`.

This fix fixes 16034.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 069663af
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment