深度学习与大模型 教案 第二章代码.docx

深度学习与大模型 教案 第二章代码.docx

第二章代码

第二章:人工神经网络与深度学习基础

importtorch

importtorch.nnasnn

importtorch.optimasoptim

importtorchvision

importtorchvision.transformsastransforms

#数据预处理

transform=transforms.Compose([

transforms.ToTensor(),

transforms.Normalize((0.5,),(0.5,))

])

#加载MNIST数据集

train_dataset=torch

您可能关注的文档

文档评论(0)

1亿VIP精品文档

相关文档