摘要: 使用基本的函数搭建一个两层神经网络(输入层不算做一层) torch.autograd包 主要用于完成神经网络后向传播中的链式求导 Variable类 torch.autograd包中的Variable类,将Tensor数据类型变量进行封装。计算图中的每个节点变为一个Variable对象,来应用自动梯 阅读全文
posted @ 2019-09-03 18:26 键盘已坏 阅读(398) 评论(0) 推荐(0) 编辑
摘要: PyTorch中的数据类型 分为浮点型和整数型两种: 初始化数组有以下四种方式 运算 torch.abs取绝对值;torch.add相加;torch.clamp裁剪;torch.div求商;torch.mul求积;torch.pow求幂;torch.mm矩阵相乘;torch.mv矩阵和向量相乘(前后 阅读全文
posted @ 2019-09-03 11:40 键盘已坏 阅读(250) 评论(0) 推荐(0) 编辑