04 2021 档案

摘要:一、简介 torch作为深度学习的主流框架,其根本在于1.具有强大的GPU加速的张量计算功能。2.包含自动求导系统的深度神经网络。自动求导功能由torch.Autograd模块实现,而科学计算部分则直接通过torch,mathfunc()来实现,本文拟对科学计算相关函数做一个简单总结。 二、torc 阅读全文

posted @ 2021-04-16 13:54 jiang_generator 阅读(207) 评论(0) 推荐(0) |

摘要:一、简介 nn全称为neural network,意思是神经网络,是torch中构建神经网络的模块。 二、子模块介绍 2.1 nn.functional 该模块包含构建神经网络需要的函数,包括卷积层、池化层、激活函数、损失函数、全连接函数等。 例如:import torch.nn.functiona 阅读全文

posted @ 2021-04-16 10:03 jiang_generator 阅读(2047) 评论(0) 推荐(0) |

博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3