获取图层权重作为网络输出keras

Question

我希望有人可以帮我解决这个问题。

我有一个自动编码器模型，我想在中间得到密集层的权重W：我必须将一些元素设置为零，并在损失函数中使用它。

基本上

x = Encoder()(input)
z = Dense(units=10)(x)
out = Decoder()(z)

并且损失应该具有以下界面

loss(input, out, z, w)

w是致密层的权重。 知道怎么样？

另外，我想强制W的对角线为零。我看到Tensorflow有一个matrix_set_diag函数，有这方面的经验吗？

Answer 1

不要抛弃对Dense（）的引用。例如：

x = Encoder()(input)
layer = Dense(units=10)
z = layer(x)
out = Decoder()(z)

print(layer.kernel)

将为您提供图层的内核