簡體   English   中英

Keras Dense 層和 Pytorch 的 nn.linear 層有區別嗎?

[英]Is there a difference between Keras Dense layer and Pytorch's nn.linear layer?

我注意到Keras 密集層的定義說:激活 function 使用。 如果您不指定任何內容,則不會應用任何激活(即“線性”激活:a(x) = x)。

所以如果我們有這樣的代碼:

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

是不是基本一樣:

nn.linear(128, 10)

?

太感謝了!

是的,如果沒有激活它只是一個線性層。

是的,它是一樣的。 model.add (Dense(10, activation = None))nn.linear(128, 10)是相同的,因為它在兩者中都沒有激活,因此如果您不指定任何內容,則不會應用激活。 原來如此:!! :)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM