diff options
| -rw-r--r-- | tensorflow/python/estimator/estimator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/estimator/estimator.py b/tensorflow/python/estimator/estimator.py index 90280fd25d..ff2baa0465 100644 --- a/tensorflow/python/estimator/estimator.py +++ b/tensorflow/python/estimator/estimator.py @@ -1653,7 +1653,7 @@ def _combine_distributed_scaffold(grouped_scaffold, distribution): def _unwrap_and_concat(value): value = nest.flatten(distribution.unwrap(value)) if len(value) != 1: - return array_ops.concat(value) + return array_ops.concat(value, 0) return value[0] ready_op = distribution.call_for_each_tower( |