logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

PyTorch 神经网络工具箱学习笔记

对比维度nn.Module用法先实例化,再以函数形式调用直接调用函数,传入数据和参数参数管理自动管理可学习参数需手动定义和传入参数与容器配合支持(如 nn.Sequential)不支持状态切换(如 dropout)自动切换(model.eval ())需手动控制(如设置 train_flag)适用场景构建可训练层(Linear、Conv2d)激活函数、池化层、损失计算这是最基础也最灵活的方式,适用

#pytorch#神经网络#学习
到底了