python
文章平均质量分 84
孤山大姐姐
这个作者很懒,什么都没留下…
展开
-
实现一个神经元的NN,支持前向、BP、及梯度计算等
本日加更!插入版权信息,完整代码在最末尾。# -*- coding: utf-8 -*-## Copyright (C) 2021 ## @Time : 2021/8/10 下午8:29# @Author : Erebor # @Email : ******@gmail.com# @File : NN.py# @Software: PyCharm基于以下两个框架,最后也没用上numpy,这个实现实在是太手残了import numpy as npimport math原创 2021-08-10 21:38:53 · 161 阅读 · 2 评论 -
简单速成Pytorch的scatter_函数理解
首先明确,这个函数实现的功能是”放“怎么个放法呢,看这个函数的参数:Tensor.scatter_(dim, index, src, reduce=None) → Tensorsrc:将src这个tensor中的值,放到self里(也就是”.“符号前面的那个Tensor)。src不一定要是一个tensor,也可以是一个值。dim及index:指示要放的具体位置。self [index[i][j][k]] [j][k] = src[i][j][k] # if dim == 0self原创 2021-08-09 17:12:36 · 1133 阅读 · 2 评论