这样写op不用写反向计算吧
收藏
用 create_parameter 创建可训练参数的方式写op,不用自己写backward吧
0
收藏
请登录后评论
框架会自己根据forward函数和loss函数算数值微分,自己进行梯度、权值更新了吧