- cos_theta = torch.mm(embbedings, kernel_norm)计算两矩阵相乘结果
- cos_theta = cos_theta.clamp(-1,1) 将input中的元素限制在[min,max]范围内并返回一个Tensor
- cos_theta_2 = torch.pow(cos_theta, 2)求幂
- torch.abs(a):各项参数的绝对值 torch.add(a,b):求和
- torch.div(a,b):求商
- torch.mul(a,b):求积(不一定是矩阵)
- torch.pow(a,b):求幂
- torch.mm(a,b):求积(按矩阵和矩阵之间的规则做)
- torch.mv(a,b):求积(按矩阵和向量之间的规则做)
人脸识别学习笔记(五):Arcface所使用的函数
最新推荐文章于 2024-04-17 21:34:47 发布