CosineSimilarity
产品支持情况
| 产品 |
是否支持 |
| Ascend 950PR/Ascend 950DT |
√ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 |
× |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
× |
| Atlas 200I/500 A2 推理产品 |
× |
| Atlas 推理系列产品 |
× |
| Atlas 训练系列产品 |
× |
功能说明
cosine_similarity(x1,x2,dim,eps)=∑(x1⋅x2,dim)max(∑(x12,dim),eps)⋅max(∑(x22,dim),eps)\text{cosine\_similarity}(x_1, x_2, \text{dim}, \text{eps}) = \frac{\sum(x_1 \cdot x_2, \text{dim})}{\max(\sqrt{\sum(x_1^2, \text{dim})}, \text{eps}) \cdot \max(\sqrt{\sum(x_2^2, \text{dim})}, \text{eps})}
cosine_similarity(x1,x2,dim,eps)=max(∑(x12,dim),eps)⋅max(∑(x22,dim),eps)∑(x1⋅x2,dim)
参数说明
| 参数名 |
输入/输出/属性 |
描述 |
数据类型 |
数据格式 |
| input_x1 |
输入 |
第一个输入张量，任意维度。 |
FLOAT |
ND |
| input_x2 |
输入 |
第二个输入张量，与input_x1同shape（或可广播）。 |
FLOAT |
ND |
| output_y |
输出 |
沿dim维度reduce后的输出张量。 |
FLOAT |
ND |
| dim |
属性 |
计算余弦相似度的维度，默认值为1。 |
INT32 |
- |
| eps |
属性 |
数值稳定性参数，防止除零，默认值为1e-8。 |
FLOAT |
- |
约束说明
- 输入张量最大支持8维。
- 输入支持float32数据类型。
- 支持广播（x1和x2 shape不同时自动广播）。
- dim属性支持负数索引（如dim=-1表示最后一维）。
调用说明
