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

相关推荐
cwn_8 分钟前
自然语言处理NLP(2)
人工智能·深度学习·神经网络·机器学习·自然语言处理
爱埋珊瑚海~~19 分钟前
开源AI智能体-JoyAgent集成Deepseek
大数据·人工智能·ai智能体
IT古董30 分钟前
【第四章:大模型(LLM)】01.神经网络中的 NLP-(2)Seq2Seq 原理及代码解析
rnn·深度学习·lstm
DogDaoDao37 分钟前
GitHub开源项目Zerox:AI驱动的OCR革命
人工智能·深度学习·开源·github·ocr·图像识别·zerox
啊阿狸不会拉杆39 分钟前
《Java 程序设计》第 8 章 - Java 常用核心类详解
java·开发语言·python·算法·intellij-idea
小白学大数据42 分钟前
Python + Requests库爬取动态Ajax分页数据
开发语言·python·ajax·okhttp
2501_9200470343 分钟前
python-内存管理
开发语言·jvm·python
Eastmount1 小时前
[AI Coding] 一.腾讯CodeBuddy IDE内测、安装及基本用法(国产AI IDE启航)
人工智能·大模型·腾讯·ai coding·codebuddy ide
意疏1 小时前
CodeBuddy的安装教程
人工智能
樱花的浪漫1 小时前
大模型推理框架基础概述
人工智能·算法·机器学习·语言模型·自然语言处理