torch.nn中functional

from torch.nn import functional as F

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

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。