PReLU具有可变输入大小

时间:2017-05-22 07:08:15

标签: tensorflow keras keras-layer keras-2

如何使用具有可变输入大小的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)

上面的代码产生

  

TypeError:*:'NoneType'和'不支持的操作数类型   'NoneType'

1 个答案:

答案 0 :(得分:0)

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