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

Cast shape to integer. Fix formatting.

Browse Source
This commit is contained in:
Trevor Morris 2019-07-10 10:13:45 -07:00
parent f74e5e1a19
commit 2f61f75e24
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/keras/layers/core.py
View File

@ -582,11 +582,11 @@ class Flatten(Layer):
input_shape = tensor_shape.TensorShape(inputs.shape).as_list()
if input_shape and all(input_shape[1:]):
outputs = array_ops.reshape(inputs, (-1, np.prod(input_shape[1:])))
outputs = array_ops.reshape(inputs, (-1, int(np.prod(input_shape[1:]))))
else:
outputs = array_ops.reshape(
inputs, (tensor_shape.dimension_value(inputs.shape[0]) or
array_ops.shape(inputs)[0], -1))
array_ops.shape(inputs)[0], -1))
if not context.executing_eagerly():
outputs.set_shape(self.compute_output_shape(inputs.shape))
return outputs