我如何在Pytorch中为backward()函数获取grad_fn可用的函数的穷尽列表?

xtupzzrd  于 12个月前  发布在  其他
关注(0)|答案(1)|浏览(114)

可以计算梯度的Tensor将具有grad_fn=AddBackward0或MulBackward0或SubBackward0。
有没有Python命令可以列出所有这些“xxxBackward0”方法及其相应的向前和向后公式?

5gfr0r5j

5gfr0r5j1#

我不知道一个单一的python命令,但你可以看看torch.Tensor的方法。
基本上所有的Tensor操作都不是“硬”的(如argsort,逻辑运算符),都有一个向后的方法实现。

相关问题