:github_url: https://github.com/pytorch/pytorch .. _program_listing_file_torch_csrc_api_include_torch_nn_modules_distance.h: Program Listing for File distance.h =================================== |exhale_lsh| :ref:`Return to documentation for file ` (``torch/csrc/api/include/torch/nn/modules/distance.h``) .. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS .. code-block:: cpp #pragma once #include #include #include #include #include #include namespace torch::nn { class TORCH_API CosineSimilarityImpl : public Cloneable { public: explicit CosineSimilarityImpl(const CosineSimilarityOptions& options_ = {}); void reset() override; void pretty_print(std::ostream& stream) const override; Tensor forward(const Tensor& input1, const Tensor& input2); CosineSimilarityOptions options; }; TORCH_MODULE(CosineSimilarity); // ============================================================================ class TORCH_API PairwiseDistanceImpl : public Cloneable { public: explicit PairwiseDistanceImpl(const PairwiseDistanceOptions& options_ = {}); void reset() override; void pretty_print(std::ostream& stream) const override; Tensor forward(const Tensor& input1, const Tensor& input2); PairwiseDistanceOptions options; }; TORCH_MODULE(PairwiseDistance); } // namespace torch::nn