Keras Dense 层和 Pytorch 的 nn.linear 层有区别吗？

Posted 2023-02-23

【中文标题】Keras Dense 层和 Pytorch 的 nn.linear 层有区别吗？

【英文标题】：Is there a difference between Keras Dense layer and Pytorch's nn.linear layer?

【发布时间】：2022-01-24 01:16:03

【问题描述】：

我注意到Keras Dense layer 的定义说：要使用的激活函数。如果您不指定任何内容，则不会应用任何激活（即“线性”激活：a(x) = x）。

如果我们有这样的代码：

model.add(Dense(10, activation = None))

是不是基本一样：

nn.linear(128, 10)

?

非常感谢！

【参考方案1】：

是的，它是一样的。 model.add (Dense(10, activation = None)) 或 nn.linear(128, 10) 是一样的，因为两者都没有激活它，因此如果你不指定任何东西，则不会应用激活。就是这样！！！ :)

【参考方案2】：

是的，如果没有激活它只是一个线性层。