推荐星级:
  • 1
  • 2
  • 3
  • 4
  • 5

线性层的张量切分逻辑

更新时间:2026-07-29 10:42:32 大小:16K 上传用户:江岚查看TA发布的资源 标签:线性切分 下载积分:2分 评价赚积分 (如何评价?) 打赏 收藏 评论(0) 举报

资料介绍

一、线性层的基本结构与张量形态

线性层(也叫全连接层)是深度学习中最基础的神经网络层,其核心计算表达式为:

 

其中各张量的维度定义通常为:

· 输入张量$X$:形状为 $ 是单次训练/推理的样本数量, 是输入特征的维度

· 权重张量$W$:形状为 $ 是输出特征的维度

· 偏置张量$b$:形状为 $,会通过广播机制和相加

· 输出张量$Y$:形状为 $

当模型规模增大,特征维度或批量尺寸超出单卡显存容量时,就需要对张量进行切分,分布式训练/推理场景下的张量切分主要分为输入特征切分输出特征切分两种核心逻辑,分别对应不同的并行策略。

部分文件列表

文件名 大小
线性层的张量切分逻辑.docx 16K

全部评论(0)

暂无评论

上传资源 上传优质资源有赏金

  • 打赏
  • 30日榜单

推荐下载