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

Avoid doing an equality check on Tensor dimensions

Browse Source 
PiperOrigin-RevId: 261015435
This commit is contained in:
Gaurav Jain 2019-07-31 15:54:43 -07:00 committed by TensorFlower Gardener
parent 1c46da48dc
commit 3351279836
1 changed files with 2 additions and 1 deletions

3
tensorflow/python/ops/special_math_ops.py
View File

@ -533,7 +533,8 @@ def _reshape_if_necessary(tensor, new_shape):
new_shape = tuple(-1 if x is None else x for x in new_shape)
cur_shape = tuple(x.value for x in tensor.get_shape().dims)
if (len(new_shape) == len(cur_shape) and
all(d0 == d1 or d1 == -1 for d0, d1 in zip(cur_shape, new_shape))):
all(not isinstance(d1, ops.Tensor) and (d0 == d1 or d1 == -1)
for d0, d1 in zip(cur_shape, new_shape))):
return tensor
else:
return array_ops.reshape(tensor, new_shape)