python - tf.layers.dense 如何创建inputs.kernel 权重矩阵？

Question

Tensorflow document :输出 = 激活(inputs.kernel + 偏差)

1. 文档指出kernel是由层创建的权重矩阵，但我在tensorflow/python/layers/core.py中找不到kernel的实现。我在哪里可以找到实现？

----------------------------------------tl;博士---------------- ----------------------------------

根据答案，权重矩阵的形状定义为 shape=[input_shape[-1].value, self.units] Line 886

Answer 1

他们已经开始将 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