趋近智
requires_grad)backward()).grad)torch.nn 搭建模型torch.nn.Module 基类torch.nn 损失)torch.optim)torch.utils.data.Datasettorchvision.transforms)torch.utils.data.DataLoader在前面介绍的基础张量操作之上,本章将讲解更高级的张量操作方法。对张量结构、数据类型和设备放置进行精细控制,对于有效准备数据和实现复杂的深度学习模型来说非常重要。
在本章中,你将学会:
view()、reshape() 和 permute() 重构张量,而不改变其数据。cat()、stack()、split() 和 chunk() 组合多个张量或将单个张量分割成不同部分。float、int),并相应地转换张量类型。.to(device) 在 CPU 内存和 GPU 加速器之间传输张量。掌握这些操作对于处理深度学习工作流中遇到的多样化数据格式和计算要求是必要的。
2.1 张量索引与切片
2.2 张量的重塑与维度调整
2.3 张量的合并与分割
2.4 理解广播机制
2.5 张量数据类型
2.6 CPU 与 GPU 张量
2.7 练习:张量操作技巧
© 2026 ApX Machine Learning用心打造