from keras.models import Model
from keras.models import Input
from keras.layers import Dense
a = Input(shape=(3,))
b = Dense(2, use_bias=False)(a)
model = Model(inputs=a, outputs=b)
假设上面代码中Dense
层的权重是[[2, 3], [3, 1], [-1, 1]]
。如果我们给[[2, 1, 3]]
作为model
的输入,那么输出将是:
但是我想将softmax函数应用到Dense
层的每一行,这样输出将是:
我该怎么做?在
实现所需内容的一种方法是通过子类化
Dense
层并重写其call
方法来定义自定义层:测试以确保其工作正常:
^{pr2}$使用numpy验证它:
相关问题 更多 >
编程相关推荐