def concatenate_layers(inputs, concat_axis, mode='concat'):
if KERAS_2:
assert mode == 'concat', "Only concatenation is supported in this wrapper"
return Concatenate(axis=concat_axis)(inputs)
else:
return merge(inputs=inputs, concat_axis=concat_axis, mode=mode)
评论列表
文章目录