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)
