Add torch library bindings (guard main, add torch wrapper functions)

ad8dab0 verified 3 days ago

124 Bytes

	#pragma once
	#include <torch/torch.h>

	std::vector<torch::Tensor> compute_kronecker(torch::Tensor ct, torch::Tensor z_inv);