tensorflow - 具有可变输入大小的 PReLU

Question

如何使用PReLU可变输入大小的激活函数？

from keras.layers import Conv2D, Input
from keras.layers.advanced_activations import PReLU

X = Input(shape=(None,None,1))
conv1Y = Conv2D(filters=49, kernel_size=7, name='conv1')(X)
conv1Y = PReLU()(conv1Y)

上面的代码产生

类型错误：* 不支持的操作数类型：“NoneType”和“NoneType”

Answer 1

使用 PreLU(shared_axes=[1,2]) 以便在编译时计算形状。