nn.Linear nn.Conv2d nn.BatchNorm2d

复制代码
conv,BN,Linear
conv:https://blog.csdn.net/Strive_For_Future/article/details/83240232
1)conv2d.weight     shape=[输出channels,输入channels,kernel_size,kernel_size]

2)conv2d.bias   shape=[输出channels]
BN:https://www.cnblogs.com/tingtin/p/12523701.html
尺寸:输入输出一样
m = nn.BatchNorm2d(2,affine=True) #2表示输出通道数,affine=True表示权重w和偏重b将被使用学习
m.weight:tensor([1., 1.], requires_grad=True)
m.bias:tensor([0., 0.], requires_grad=True)#w,b都是大小维输出通道数的向量
Linear:https://www.cnblogs.com/tingtin/p/12425849.html
nn.Linear()用于设置全连接层,输入输出均为二维张量,形状为[batch_size, size]
def __init__(self, in_features, out_features, bias=True):
        super(Linear, self).__init__()
        self.in_features = in_features
        self.out_features = out_features
        self.weight = Parameter(torch.Tensor(out_features, in_features))
        if bias:
            self.bias = Parameter(torch.Tensor(out_features))
        else:
            self.register_parameter('bias', None)
        self.reset_parameters()
复制代码

 

posted on   cltt  阅读(547)  评论(0编辑  收藏  举报

编辑推荐:
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

导航

统计

点击右上角即可分享
微信分享提示