Tensorflow文档: outputs = activation(inputs.kernel + bias)
- 文档中提到
kernel
是由层创建的权重矩阵,但我无法在tensorflow/python/layers/core.py中找到kernel
的实现。请问在哪里可以找到这个实现?
—————————-tl;dr————————————-
根据回答,权重矩阵的形状定义为shape=[input_shape[-1].value, self.units]
第886行
回答:
他们已经开始混合使用tf.layers
和tf.keras
。在你引用的实现中,该类继承了来自keras_layers
的所有方法
“顺藤摸瓜”,实现位于这里,其核心内容如下:
def build(...): self.kernel = self.add_variable('kernel', ... self.bias = self.add_variabledef call(...): # ... outputs = gen_math_ops.mat_mul(inputs, self.kernel) # ... if self.activation is not None: return self.activation(outputs) return outputs