时间戳

首页 > 解决方案 > tf.layers.dense 如何创建 inputs.kernel 权重矩阵？

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

问题描述

张量流文档： outputs = activation(inputs.kernel + bias)

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

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

从答案来看，权重矩阵的形状定义为shape=[input_shape[-1].value, self.units] 第 886 行

标签： pythontensorflowmachine-learning

解决方案

他们已经开始tf.layers与tf.keras. 在您引用的实现中，该类继承了所有方法keras_layers

“跟着钱走”，执行就到这里，归结为

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

推荐阅读