| 192 | |
| 193 | @classmethod |
| 194 | def setUpClass(cls): |
| 195 | print("-" * 20, "Layer_QuanDense_Test", "-" * 20) |
| 196 | cls.batch_size = 4 |
| 197 | cls.inputs_shape = [cls.batch_size, 10] |
| 198 | |
| 199 | cls.ni = Input(cls.inputs_shape, name='input_layer') |
| 200 | cls.layer1 = QuanDense(n_units=5) |
| 201 | nn = cls.layer1(cls.ni) |
| 202 | cls.layer1._nodes_fixed = True |
| 203 | cls.M = Model(inputs=cls.ni, outputs=nn) |
| 204 | |
| 205 | cls.layer2 = QuanDense(n_units=5, in_channels=10) |
| 206 | cls.layer2._nodes_fixed = True |
| 207 | |
| 208 | cls.inputs = tf.random.uniform((cls.inputs_shape)) |
| 209 | cls.n1 = cls.layer1(cls.inputs) |
| 210 | cls.n2 = cls.layer2(cls.inputs) |
| 211 | cls.n3 = cls.M(cls.inputs, is_train=True) |
| 212 | |
| 213 | print(cls.layer1) |
| 214 | print(cls.layer2) |
| 215 | |
| 216 | @classmethod |
| 217 | def tearDownClass(cls): |