experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Changes keep_dims to keepdims to remove deprecation warning.

Browse Source 
#labeledtensor

PiperOrigin-RevId: 186071210
This commit is contained in:
A. Unique TensorFlower 2018-02-16 17:55:07 -08:00 committed by TensorFlower Gardener
parent bde6b79a2d
commit 0c14cf398c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/labeled_tensor/python/ops/ops.py
View File

@ -951,7 +951,7 @@ def define_reduce_op(op_name, reduce_fn):
intermediate_axes.append(axis)
reduce_op = reduce_fn(
labeled_tensor.tensor, reduction_dimensions, keep_dims=True)
labeled_tensor.tensor, reduction_dimensions, keepdims=True)
reduce_lt = core.LabeledTensor(reduce_op, intermediate_axes)
return squeeze(reduce_lt, axes_to_squeeze, name=scope)