| 17 | |
| 18 | @classmethod |
| 19 | def setUpClass(cls): |
| 20 | print("-" * 20, "Layer_Stack_Test", "-" * 20) |
| 21 | cls.batch_size = 4 |
| 22 | cls.inputs_shape = [cls.batch_size, 10] |
| 23 | |
| 24 | cls.ni = Input(cls.inputs_shape, name='input_layer') |
| 25 | a = Dense(n_units=5)(cls.ni) |
| 26 | b = Dense(n_units=5)(cls.ni) |
| 27 | cls.layer1 = Stack(axis=1) |
| 28 | cls.n1 = cls.layer1([a, b]) |
| 29 | cls.M = Model(inputs=cls.ni, outputs=cls.n1) |
| 30 | |
| 31 | cls.inputs = tf.random.uniform(cls.inputs_shape) |
| 32 | cls.n2 = cls.M(cls.inputs, is_train=True) |
| 33 | |
| 34 | print(cls.layer1) |
| 35 | |
| 36 | @classmethod |
| 37 | def tearDownClass(cls): |