Tensor 是什么?PyTorch 里最重要的对象讲清楚
Tensor 可以理解成更通用的数组,它承载数据、参数和梯度,是 PyTorch 训练流程的基础。
和「PyTorch」相关的文章都收在这里,方便沿着同一个主题继续读下去。
同一主题下的内容,会更容易看出思路是怎么慢慢长出来的。
Tensor 可以理解成更通用的数组,它承载数据、参数和梯度,是 PyTorch 训练流程的基础。
Linear 层本质上是一次线性变换,它把输入特征通过权重矩阵和偏置映射到新的输出空间。
DataLoader 负责把数据按批次送进模型,并处理 shuffle、batch size 和多进程加载等训练细节。
从准备数据、定义模型、选择损失函数和优化器,到训练循环和验证评估,把 PyTorch 训练主线串起来。