diff options
author | 2018-09-17 02:14:03 -0700 | |
---|---|---|
committer | 2018-09-17 02:14:07 -0700 | |
commit | b1f4328517851e76cff3d4af8766e7e3446314ba (patch) | |
tree | 8a1be4536120c907908646911fa051575a1657dd | |
parent | 5ff7f982846bd3f8056c8252a0afeb07e5b3e982 (diff) | |
parent | 72359f9cfa10a08cecc3a179999a1b8ab835a818 (diff) |
Merge pull request #22289 from wangsiyu:estimator_concat_fix
PiperOrigin-RevId: 213236122
-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( |