tf.layers.dense如何创建inputs.kernel权重矩阵?

时间:2018-08-01 05:08:48

标签: python tensorflow machine-learning

Tensorflow documentoutputs = activation(inputs.kernel + bias)

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

---------------------------- tl; dr ---------------- ---------------------

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

1 个答案:

答案 0 :(得分:2)

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