当前位置: 首页 > news >正文

pytorch基础-nn.linear

import torch
import torch.nn as nn

# 定义线性层
linear_layer = nn.Linear(in_features=10, out_features=5, bias=True)

# 输入数据
input_data = torch.randn(32, 10)  # (batch_size=32, in_features=10)

# 前向传播
output = linear_layer(input_data)
print(output.shape)  # 输出形状: (32, 5)

 维度变化

  • 输入(batch_size, in_features)

  • 输出(batch_size, out_features)

示例

  • 输入形状:(32, 10)

  • 线性层:nn.Linear(10, 5)

  • 输出形状:(32, 5)

实现细节:矩阵乘法

http://www.dtcms.com/a/37378.html

相关文章:

  • 【软件测试阶段一】【软件测试:确保项目质量的关键防线】
  • c++day5
  • Spring Boot + Vue 接入腾讯云人脸识别API(SDK版本3.1.830)
  • 【Elasticsearch】使用Postman操作es的_bulk批量操作
  • Redis面试知识点总结
  • 【原创工具】文件清单生成器 By怜渠客
  • 【算法设计与分析】(一)介绍算法与复杂度分析
  • WEEX交易所安全教學:如何應對剪切板被劫持駭客攻擊?
  • 复合字面量
  • 第十八:路由传参 query
  • Jupyter Notebook中使用GPU进行计算
  • ubuntu离线安装Ollama并部署Llama3.1 70B INT4
  • 刷题总结 栈和队列:单调栈
  • 玩机日记 14 飞牛fnOS部署qBittorrent、AList、Jellyfin,实现下载、存取、刮削、观看一体的家庭影音中心
  • 基于 CFD 预测的机器学习第 2 部分:在 Benchmark 应用程序上使用 Stochos 预测流场
  • VMware建立linux虚拟机
  • github 部署前端静态网页(react vite)
  • 【PLL】相位检测器:PFD
  • Java多线程安全
  • QT零基础学习之路(四)--信号和槽机制
  • 第五章:队列管理模块
  • DAY40|动态规划Part08|LeetCode: 121. 买卖股票的最佳时机 、 122.买卖股票的最佳时机II 、 123.买卖股票的最佳时机III
  • 文本编辑器使用指南:Linux中的文本编辑器大冒险
  • 算法题(76):跳跃游戏II
  • 【JavaWeb13】了解ES6的核心特性,对于提高JavaScript编程效率有哪些潜在影响?
  • 静止的钉子
  • transformer架构嵌入层位置编码之动态NTK-aware位置编码
  • 第四章 哈希表
  • 每天一个Flutter开发小项目 (4) : 构建收藏地点应用 - 深入Flutter状态管理
  • 递归、搜索与回溯算法 —— 名词解析