def masking(layer, layer_in, layerId):
out = {layerId: Masking(mask_value=layer['params']['mask_value'])(*layer_in)}
return out
# ********** Convolution Layers **********
微信
公众号
扫码关注公众号