nn.Linear() 使用提醒

原本以为它是和nn.Conv2d()一样,就看第二个维度的数值,今天才知道,它是只看最后一个维度的数值!!!

例子1

python 复制代码
'''
Descripttion: 
Result: 
Author: Philo
Date: 2024-02-27 14:33:50
LastEditors: Philo
LastEditTime: 2024-02-27 14:36:22
'''
import torch
import torch.nn as nn

# 定义输入特征数量和输出特征数量
in_features = 10
out_features = 5
# 创建一个线性变换层
linear_layer = nn.Linear(in_features, out_features)
# 随机生成一个输入张量
input_tensor = torch.randn(1, 12,32,10)
# 对输入张量进行线性变换
output_tensor = linear_layer(input_tensor)
# 打印输出张量
print("输出张量的形状:", output_tensor.shape)

输出张量的形状: torch.Size([1, 12, 32, 5])

例子2

例子3 Conv2d

相关推荐
aneasystone本尊14 分钟前
重温 Java 21 学习笔记
人工智能
zandy101130 分钟前
架构解析:衡石科技如何基于AI+Data Agent重构智能数据分析平台
人工智能·科技·架构
golang学习记1 小时前
免费解锁 Cursor AI 全功能:4 种开发者私藏方案揭秘
人工智能
Andy_shenzl1 小时前
31、LangChain开发框架(八)-- LangChain 数据分析智能体实战
数据挖掘·数据分析·langchain·agent开发
图扑软件1 小时前
热力图可视化为何被广泛应用?| 图扑数字孪生
大数据·人工智能·信息可视化·数字孪生·可视化·热力图·电力能源
qq_436962181 小时前
AI驱动数据分析革新:奥威BI一键生成智能报告
人工智能·信息可视化·数据分析
广州明周科技1 小时前
Revit 200+新功能之“房间面积图表”,房间数据可视化功能,轻松洞察项目空间信息!
ai·信息可视化·数据分析·bim·revit二次开发·revit·deepseek
成长痕迹1 小时前
【MATLAB 数据分析学习指南】
matlab·信息可视化·数据分析
卓码软件测评1 小时前
借助大语言模型实现高效测试迁移:Airbnb的大规模实践
开发语言·前端·javascript·人工智能·语言模型·自然语言处理
小白狮ww1 小时前
dots.ocr 基于 1.7B 参数实现多语言文档处理,性能达 SOTA
人工智能·深度学习·机器学习·自然语言处理·ocr·小红书·文档处理