可以计算梯度的Tensor将具有grad_fn=AddBackward0或MulBackward0或SubBackward0。有没有Python命令可以列出所有这些“xxxBackward0”方法及其相应的向前和向后公式?
5gfr0r5j1#
我不知道一个单一的python命令,但你可以看看torch.Tensor的方法。基本上所有的Tensor操作都不是“硬”的(如argsort,逻辑运算符),都有一个向后的方法实现。
1条答案
按热度按时间5gfr0r5j1#
我不知道一个单一的python命令,但你可以看看torch.Tensor的方法。
基本上所有的Tensor操作都不是“硬”的(如argsort,逻辑运算符),都有一个向后的方法实现。