torch.nn中functional

from torch.nn import functional as F

nn中還有一個很常用的模塊：nn.functional，nn中的大多數layer，在functional中都有一個與之相對應的函數。nn.functional中的函數和nn.Module的主要區別在于，用nn.Module實現的layers是一個特殊的類，都是由class layer(nn.Module)定義，會自動提取可學習的參數。而nn.functional中的函數更像是純函數，由def function(input)定義。下面舉例說明functional的使用，并指出二者的不同之處。

import torch as t from torch import nn input = t.randn(2,

總結

以上是生活随笔為你收集整理的pytorch functional的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。