(input_shape)
| 78 | print('-' * 20, 'test_net2', '-' * 20) |
| 79 | |
| 80 | def get_unstack_model(input_shape): |
| 81 | ni = Input(input_shape) |
| 82 | |
| 83 | nn = Dropout(keep=0.9)(ni) |
| 84 | |
| 85 | a, b, c = UnStack(axis=-1)(nn) |
| 86 | |
| 87 | b = Flatten()(b) |
| 88 | b = Dense(10)(b) |
| 89 | |
| 90 | c = Flatten()(c) |
| 91 | |
| 92 | M = Model(inputs=ni, outputs=[a, b, c]) |
| 93 | return M |
| 94 | |
| 95 | net = get_unstack_model([None, 24, 24, 3]) |
| 96 |