diff options
| author |  wangsiyu <siyu.wsy@gmail.com> | 2018-09-15 16:02:22 +0800 |
|---|---|---|
| committer | wangsiyu <siyu.wsy@gmail.com> | 2018-09-15 16:02:22 +0800 |
| commit | 72359f9cfa10a08cecc3a179999a1b8ab835a818 (patch) | |
| tree | 82332aba72b47e705db5bb6fe08b1d7e166d577b /tensorflow/python/estimator | |
| parent | 7d4773c0d68fb5f270aaecd1954e11352acb1be8 (diff) | |
fix bug of lacking axis when using array.ops.concat in unwrap_and_concat function
Diffstat (limited to 'tensorflow/python/estimator')
| -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( |