3 #include <torch/csrc/autograd/function.h> 4 #include <torch/csrc/autograd/variable.h> 6 #include <ATen/TensorGeometry.h> 8 #include <c10/util/Optional.h> 13 namespace torch {
namespace autograd {
16 variable_list apply(variable_list&& grads)
override;
33 std::shared_ptr<Function> fn_);
35 variable_list apply(variable_list&& inputs)
override;
36 void release_variables()
override;
40 std::shared_ptr<Function> fn;
Represents a a compute device on which a tensor is located.
Variable A Variable augments a Tensor with the ability to interact in our autograd machinery...