Typedef torch::nn::functional::CosineSimilarityFuncOptions

• Defined in File distance.h

Typedef Documentation

using torch::nn::functional::CosineSimilarityFuncOptions = CosineSimilarityOptions

Options for torch::nn::functional::cosine_similarity.

See the documentation for torch::nn::CosineSimilarityOptions class to learn what arguments are supported.

Example:

namespace F = torch::nn::functional;
F::cosine_similarity(input1, input2,
F::CosineSimilarityFuncOptions().dim(1));