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

相关推荐
MongoVIP34 分钟前
Scrapy爬虫实战:正则高效解析豆瓣电影
python·scrapy
李小白6639 分钟前
Python文件操作
开发语言·python
飞哥数智坊1 小时前
3B参数差点干翻32B模型,Qwen3 Next 是如何做到的?
人工智能
人工智能技术派1 小时前
Whisper推理源码解读
人工智能·语言模型·whisper·语音识别
YangYang9YangYan1 小时前
金融分析师技能提升路径与学习资源指南
金融·数据分析
weixin_525936332 小时前
金融大数据处理与分析
hadoop·python·hdfs·金融·数据分析·spark·matplotlib
编码追梦人2 小时前
AI 重塑行业格局:从金融风控到智能制造的深度实践
人工智能·制造
Lululaurel2 小时前
提示工程深度解析:驾驭大语言模型的艺术与科学
人工智能·ai·aigc·提示词
Zwb2997922 小时前
Day 30 - 错误、异常与 JSON 数据 - Python学习笔记
笔记·python·学习·json
simon_skywalker2 小时前
第7章 n步时序差分 n步时序差分预测
人工智能·算法·强化学习