神经网络常见操作(卷积)输入输出

卷积

  • dim=d的tensor可以进行torch.nn.Convnd(in_channels,out_channels),其中n=d-1,d-2
  • 对于torch.nn.Convnd(in_channels,out_channels),改变的是tensor的倒数n+1维的大小

全连接

  • 使用torch.nn.Linear(in_features,out_features,bias)实现
  • Y=XWT +b,其中X 的形状为 (batch_size,in_features),而 W T的形状为 (in_features,out_features),b 的形状为 (out_features),在加法操作中会自动扩展为 (batch_size,out_features)。
  • 在 PyTorch 中,torch.nn.Linear 的权重矩阵W不需要手动设置,它会在初始化时自动生成并进行默认的初始化,使用均匀分布或正态分布(由参数设定)
  • 偏置向量 b 默认为全零
相关推荐
m沐沐17 小时前
【深度学习】dlib 人脸关键点检测
人工智能·深度学习·opencv·计算机视觉·pycharm·人脸识别·关键点检测
逻辑君18 小时前
认知神经科学研究报告【20260112】
人工智能·深度学习·数学建模·量子计算
Lambert-Wu19 小时前
常用激活函数总结
人工智能·深度学习·机器学习
ForDreamMusk19 小时前
批量归一化
人工智能·算法·机器学习
秉烛夜读1 天前
机器学习:基本术语
机器学习
卡梅德生物科技小能手1 天前
卡梅德生物:一文读懂核心炎症靶点TNF(肿瘤坏死因子)
经验分享·深度学习·生活
依然范特东1 天前
动手学深度学习笔记--数据操作、线性代数
人工智能·笔记·深度学习
感性的随1 天前
【机器人 / 强化学习】LWD(Learning while Deploying)机器人持续进化的强化学习框架
人工智能·深度学习·机器人
wahahaman1 天前
基于GBDT的次日降水量预测实验
人工智能·python·机器学习·数据分析
依然范特东1 天前
深度学习入门学习笔记--感知机、神经网络
笔记·深度学习·学习