def test_merge_mul(self): b1 = BLayer.InputLayer(input_shape=(3, 5)) b2 = BLayer.InputLayer(input_shape=(3, 5)) blayer = BLayer.Merge(layers=[b1, b2], 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_newapi(klayer, blayer, input_data)
def test_merge_max(self): b1 = BLayer.InputLayer(input_shape=(2, 5, 8)) b2 = BLayer.InputLayer(input_shape=(2, 5, 8)) blayer = BLayer.Merge(layers=[b1, b2], 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_newapi(klayer, blayer, input_data)