diff options
Diffstat (limited to 'tensorflow/g3doc/how_tos/adding_an_op/zero_out_grad_2.py')
-rw-r--r-- | tensorflow/g3doc/how_tos/adding_an_op/zero_out_grad_2.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/g3doc/how_tos/adding_an_op/zero_out_grad_2.py b/tensorflow/g3doc/how_tos/adding_an_op/zero_out_grad_2.py index 9734c0ce96..dc24678e33 100644 --- a/tensorflow/g3doc/how_tos/adding_an_op/zero_out_grad_2.py +++ b/tensorflow/g3doc/how_tos/adding_an_op/zero_out_grad_2.py @@ -40,5 +40,5 @@ def _zero_out_grad(op, grad): shape = array_ops.shape(to_zero) index = array_ops.zeros_like(shape) first_grad = array_ops.reshape(grad, [-1])[0] - to_zero_grad = sparse_ops.sparse_to_dense(index, shape, first_grad, 0) + to_zero_grad = sparse_ops.sparse_to_dense([index], shape, first_grad, 0) return [to_zero_grad] # List of one Tensor, since we have one input |