We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 71beb6b commit 76c3842Copy full SHA for 76c3842
src/pytti/tensor_tools.py
@@ -65,6 +65,7 @@ def format_module(module, dest, *args, **kwargs) -> torch.tensor:
65
return format_input(output, module, dest)
66
67
68
+# https://pytorch.org/docs/stable/autograd.html#function
69
class ReplaceGrad(torch.autograd.Function):
70
"""
71
returns x_forward during forward pass, but evaluates derivates as though
0 commit comments