Tensorflow document:
outputs = activation(inputs.kernel + bias)
kernel
是由图层创建的权重矩阵,但是在tensorflow/python/layers/core.py中找不到kernel
的实现。我在哪里可以找到实现?---------------------------- tl; dr ---------------- ---------------------
从答案中,权重矩阵的形状定义为shape=[input_shape[-1].value, self.units]
Line 886
答案 0 :(得分:2)
他们已经开始将tf.layers
与tf.keras
混合使用。在您引用的实现中,该类继承了keras_layers
“跟随金钱”,实现为here,归结为
def build(...):
self.kernel = self.add_variable('kernel', ...
self.bias = self.add_variable
def call(...):
# ...
outputs = gen_math_ops.mat_mul(inputs, self.kernel)
# ...
if self.activation is not None:
return self.activation(outputs)
return outputs