Similarity.h

Go to the documentation of this file.

#pragma once
 
#include "Common.h"
 
namespace reg23 {
 
std::tuple<at::Tensor, double, double, double, double, double> NormalisedCrossCorrelation_CPU(
    const at::Tensor &a, const at::Tensor &b);
 
__host__ std::tuple<at::Tensor, double, double, double, double, double> NormalisedCrossCorrelation_CUDA(
    const at::Tensor &a, const at::Tensor &b);
 
struct Similarity {
 
    __host__ static void Common(const at::Tensor &a, const at::Tensor &b, at::DeviceType device) {
        // a and b should contain floats or doubles, match in size and be on the chosen device
        TORCH_CHECK(a.sizes() == b.sizes())
        TORCH_CHECK(a.dtype() == b.dtype())
        TORCH_CHECK(a.dtype() == at::kFloat)
        TORCH_INTERNAL_ASSERT(a.device().type() == device)
        TORCH_INTERNAL_ASSERT(b.device().type() == device)
    }
};
 
} // namespace reg23