def test_merge_mul(self): z1 = ZLayer.InputLayer(input_shape=(3, 5)) z2 = ZLayer.InputLayer(input_shape=(3, 5)) zlayer = ZLayer.Merge(layers=[z1, z2], mode="mul") k1 = KLayer.InputLayer(input_shape=(3, 5)) k2 = KLayer.InputLayer(input_shape=(3, 5)) klayer = KLayer.Merge(layers=[k1, k2], mode="mul") input_data = [np.random.random([2, 3, 5]), np.random.random([2, 3, 5])] self.compare_layer(klayer, zlayer, input_data)
def test_merge_max(self): z1 = ZLayer.InputLayer(input_shape=(2, 5, 8)) z2 = ZLayer.InputLayer(input_shape=(2, 5, 8)) zlayer = ZLayer.Merge(layers=[z1, z2], mode="max") k1 = KLayer.InputLayer(input_shape=(2, 5, 8)) k2 = KLayer.InputLayer(input_shape=(2, 5, 8)) klayer = KLayer.Merge(layers=[k1, k2], mode="max") input_data = [ np.random.random([3, 2, 5, 8]), np.random.random([3, 2, 5, 8]) ] self.compare_layer(klayer, zlayer, input_data)