torch学习笔记(二) nn类结构-Linear

news/2025/1/16 0:48:19/

Linear 是module的子类,是参数化module的一种,与其名称一样,表示着一种线性变换。

创建

parent 的init函数

Linear的创建需要两个参数,inputSize 和 outputSize
inputSize:输入节点数
outputSize:输出节点数
所以Linear 有7个字段:

weight : Tensor , outputSize ×× inputSize
bias: Tensor ,outputSize
gradWeight: Tensor , outputSize ×× inputSize
gradBias: Tensor ,outputSize
gradInput: Tensor
output: Tensor
_type: output:type()
例子
module = nn.Linear(10, 5)
1
Forward Pass

---------------------
作者:bubbleoooooo
来源:CSDN
原文:https://blog.csdn.net/u012936765/article/details/52671156
版权声明:本文为博主原创文章,转载请附上博文链接!


http://www.ppmy.cn/news/607437.html

相关文章

大数据必学Java基础(三十五):深入了解关键词this

文章目录 深入了解关键词this 一、创建对象的过程 二、this关键字用法

AI训练与推理芯片

AI训练与推理芯片 训练芯片 1.1.云燧T20 基于邃思2.0芯片打造的面向数据中心的第二代人工智能训练加速卡,具有模型覆盖面广、性能强、软件生态开放等特点,可支持多种人工智能训练场景。同时具备灵活的可扩展性,提供业界领先的人工智能算力集…

机器翻译注意力机制及其PyTorch实现

前面阐述注意力理论知识,后面简单描述PyTorch利用注意力实现机器翻译 Effective Approaches to Attention-based Neural Machine Translation 简介 转存失败重新上传取消转存失败重新上传取消转存失败重新上传取消 Attention介绍 在翻译的时候,选择性的选…

大数据必学Java基础(三十六):深入了解关键词static

文章目录 深入了解关键词static 一、static修饰属性 二、static修饰方法 深入了解关键词static static可以修饰:属性,方法,代码块,内部类。

[JS][编程题]括号匹配

括号匹配 题目来源:牛客网 题目链接:括号匹配 时间限制:C/C 1秒,其他语言2秒 空间限制:C/C 256M,其他语言512M 题目描述 给定一个只包含括号的字符串,判断字符串是否有效。其中,括…

芯片封测技术

芯片封测技术 长电科技是全球领先的集成电路制造和技术服务提供商,提供全方位的芯片成品制造一站式服务,包括集成电路的系统集成、设计仿真、技术开发、产品认证、晶圆中测、晶圆级中道封装测试、系统级封装测试、芯片成品测试并可向世界各地的半导体客户…

Unity的小工具

最近项目工期紧,所以更新少了,不过我遇到一些问题,当我的UI/序列帧过多的时候,我需要去选中UI转换成Sprite,而且用shift选中的时候,文件夹中上百个文件,中间混进一个其他格式的文件我还得一个一…