2021年8月17日
Pytorch使用autograd.function自定义op
摘要： 0x01 简介 Pytorch是利用Tensor与Function来构建计算图的，Function就像计算图中的边，实现Tensor的计算，并输出新的Tensor，因此Function需要有forward和backward的功能,前者用于正常传递输入,后者用于计算梯度.(但直接用nn.Module构 阅读全文
posted @ 2021-08-17 01:02 Valeyard 阅读(413) 评论(0) 推荐(0) 编辑