cahlen
/

class-numbers-cuda

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

a51a509 verified 2 days ago

124 Bytes

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

	std::vector<torch::Tensor> compute_class_numbers_torch(torch::Tensor discriminants);