Function at::cosine_embedding_loss#

• Defined in File Functions.h

Function Documentation#

inline at::Tensor at::cosine_embedding_loss(const at::Tensor &input1, const at::Tensor &input2, const at::Tensor &target, double margin = 0.0, int64_t reduction = at::Reduction::Mean)#