keras Lambda函数的应用

自定义一个函数：

def pad_backend(inputs, in_channels, out_channels):
    pad_dim = (out_channels - in_channels)//2
    return K.spatial_3d_padding(inputs, padding=((0, 0), (0, 0), (pad_dim, pad_dim)))

使用这个函数，参数以字典形式传递，默认第一个参数是tensor：

identity = Lambda(pad_backend,arguments={'in_channels':in_channels, 'out_channels':out_channels})(identity)