pytorch nn.Linear

最新推荐文章于 2023-08-15 10:22:32 发布

明朝百晓生

最新推荐文章于 2023-08-15 10:22:32 发布

阅读量655

点赞数

分类专栏：人工智能文章标签： pytorch 深度学习人工智能

4AM_明朝百晓生

本文链接：https://blog.csdn.net/chengxf2/article/details/128012561

版权

人工智能专栏收录该内容

102 篇文章 3 订阅

订阅专栏

1：参数说明

2：功能

3：例子

参考： nn.Linear的理解 - 知乎

一参数说明：

torch.nn.Linear(

in_features, 输入的神经元个数

out_features, 输出神经元个数

bias=True 是否包含偏置 )

模型主要实现如下功能：（x,w 都是行向量）

二实现功能：

$y=xw^T+b$

向量是按照行向量排列的

$\frac{\partial y}{\partial w}=x$

三：例子

# -*- coding: utf-8 -*-
"""
Created on Thu Nov 24 09:25:45 2022

@author: chengxf2
"""
import torch 
from torch import nn


def  Linear():
    
     conv =  nn.Linear(in_features= 4, out_features =2)
     
     X = torch.randn(10, 4)    
     
     out = conv(X)
     
     print("\n 输出 ",out.shape)
     print("\n 权重系数",conv.weight.shape)
     print("\n 偏置系数 ",conv.bias.shape)
  
    
if __name__ == "__main__":
    
    Linear()

输出 torch.Size([10, 2])

权重系数 torch.Size([2, 4])

偏置系数 torch.Size([2])