Crate caffe2op_cos
source ·Structs
- | Calculates the cosine of the given input | tensor, element-wise. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/cos_op.cc |
- | CosineEmbeddingCriterion takes two inputs: the | similarity value and the label, and computes the | elementwise criterion output as | | output = 1 - s, if y == 1 | max(0, s - margin), if y == -1