Address ebrevdo

ShiSanChuan · Aug 10, 2018 · dbfa330 · dbfa330
1 parent b2470ca
commit dbfa330
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensorflow/python/ops/nn_grad.py b/tensorflow/python/ops/nn_grad.py
@@ -987,8 +987,7 @@ def _TopKGrad(op, grad, _):
   # Compute linear indices (flattened to 1D).
   ind = array_ops.reshape(ind_2d + math_ops.cast(array_ops.expand_dims(
       math_ops.range(0, math_ops.cast(outerdim, dtypes.int64)
-                     * in_lastdim, in_lastdim), -1
-  ), dtypes.int32), [-1])
+                     * in_lastdim, in_lastdim), -1), dtypes.int32), [-1])
 
   # Substitute grad to appropriate locations and fill the rest with zeros,
   # finally reshaping it to the original input shape.