Tensorflow和Numpy产生不同的结果

时间:2020-05-24 22:47:20

标签: python numpy tensorflow keras

我已经看到here,Tensorflow在Dense层中使用了matmul。 我尝试在Numpy中执行相同的操作,但是会产生不同的结果。

if __name__ == "__main__":

1 个答案:

答案 0 :(得分:2)

您对该操作的理解是正确的,并且您的代码几乎是正确的。 尝试更换

yy = tf.keras.layers.Dense(128, activation='relu', weights=[w], use_bias=False)

使用

yy = tf.keras.layers.Dense(128, activation=None, kernel_initializer=lambda *args, **kwargs: w, use_bias=False)

以防止权重的随机初始化,并且测试将通过。