Hi, I ran the example code but got errors:
trimmed_output.backward(trimmed_grad)
File "/usr/local/lib64/python3.6/site-packages/torch/tensor.py", line 185, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/usr/local/lib64/python3.6/site-packages/torch/autograd/init.py", line 127, in backward
allow_unreachable=True) # allow_unreachable flag
RuntimeError: module 'cpp_functions' has no attribute 'backward_input'
Any helps, please?