#pragma once #include #include namespace torch { namespace autograd { using GradMode = at::GradMode; using AutoGradMode = at::AutoGradMode; }}