Go to the source code of this file.
|
| std::tuple< at::Tensor, double, double, double, double, double > | reg23::NormalisedCrossCorrelation_CPU (const at::Tensor &a, const at::Tensor &b) |
| | Additionally returns intermediate quantities useful for evaluating the backward pass.
|
| |
| __host__ std::tuple< at::Tensor, double, double, double, double, double > | reg23::NormalisedCrossCorrelation_CUDA (const at::Tensor &a, const at::Tensor &b) |
| | An implementation of reg23::NormalisedCrossCorrelation_CPU that uses CUDA parallelisation.
|
| |