Changed to shape from get_shape()
Used the recommended method
This commit is contained in:
parent
8f75c5aa21
commit
9c8e727c59
@ -523,8 +523,8 @@ class TopologyConstructionTest(keras_parameterized.TestCase):
|
|||||||
self.assertEqual(len(model.inputs), 2)
|
self.assertEqual(len(model.inputs), 2)
|
||||||
self.assertEqual(g.name, 'model/dense_2/BiasAdd:0')
|
self.assertEqual(g.name, 'model/dense_2/BiasAdd:0')
|
||||||
|
|
||||||
self.assertListEqual(g.shape.as_list(), c.get_shape().as_list())
|
self.assertListEqual(g.shape.as_list(), c.shape.as_list())
|
||||||
self.assertListEqual(h.shape.as_list(), d.get_shape().as_list())
|
self.assertListEqual(h.shape.as_list(), d.shape.as_list())
|
||||||
|
|
||||||
# test separate manipulation of different layer outputs
|
# test separate manipulation of different layer outputs
|
||||||
i = keras.layers.Dense(7, name='dense_4')(h)
|
i = keras.layers.Dense(7, name='dense_4')(h)
|
||||||
|
Loading…
Reference in New Issue
Block a user