torch.nn.Linear()和torch.nn.functional.linear()如何使用

本篇文章給大家分享的是有關torch.nn.Linear()和torch.nn.functional.linear()如何使用，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

代碼實驗展示:

Microsoft Windows [版本 10.0.18363.1316](c) 2019 Microsoft Corporation。保留所有權利。

C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> import torch.nn as nn>>> import torch.nn.functional as F>>> torch.manual_seed(seed=20200910)<torch._C.Generator object at 0x000002470C22D330>>>>>>> data_in = torch.randn(3,4,5,6,7)>>> data_in.shape
torch.Size([3, 4, 5, 6, 7])>>> linear4cxq = nn.Linear(7, 30)>>>>>> linear4cxq
Linear(in_features=7, out_features=30, bias=True)>>>>>> data_out = linear4cxq(data_in)>>> data_out.shape
torch.Size([3, 4, 5, 6, 30])>>>>>> data_in.shape
torch.Size([3, 4, 5, 6, 7])>>>>>>>>>

以上就是torch.nn.Linear()和torch.nn.functional.linear()如何使用，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。