SurajK7

#include <torch/torch.h>
#include <torch/csrc/autograd/variable.h>
#include <torch/csrc/autograd/function.h>
#include <torch/csrc/jit/frontend/tracer.h>
#include <torch/csrc/jit/ir/ir.h>
#include <torch/csrc/jit/api/module.h>
#include <torch/csrc/jit/passes/fixup_trace_scope_blocks.h>
#include <torch/script.h> // One-stop header.
#include <vector>
#include <iostream>